f6ceb97a6199dd781fb88d04de556df5dbd79b8e,kymatio/scattering2d/backend/backend_skcuda.py,Pad,build,#Pad#,68

Before Change

        self.build()

    def build(self):
        self.padding_module = ReflectionPad2d(self.pad_size)

    def __call__(self, x):
        if not self.pre_pad:
            x = self.padding_module(x)

After Change

        if pad_size_tmp[2] == self.input_size[1]:
            pad_size_tmp[2] -= 1
            pad_size_tmp[3] -= 1
        self.padding_module = ReflectionPad2d(pad_size_tmp)

    def __call__(self, x):
        if not self.pre_pad:
            x = self.padding_module(x)

Instances