def __init__(self, M, N, J, pre_pad=False, jit=True):
super(Scattering, self).__init__()
self.M, self.N, self.J = M, N, J
self.pre_pad = pre_pad
self.jit = jit
self.fft = Fft()
self.modulus = Modulus(jit=jit)
self.periodize = Periodize(jit=jit)
self._prepare_padding_size([1, 1, M, N])
self.padding_module = pad_function(2**J)
# Create the filters
filters = filters_bank(self.M_padded, self.N_padded, J)
self.Psi = filters['psi']
self.Phi = [filters['phi'][j] for j in range(J)]
def __init__(self, M, N, J, pre_pad=False, jit=True):
super(Scattering, self).__init__()
self.M, self.N, self.J = M, N, J
self.pre_pad = pre_pad
self.jit = jit
self.fft = Fft()
self.modulus = Modulus(jit=jit)
self.periodize = Periodize(jit=jit)
self._prepare_padding_size([1, 1, M, N])
self.padding_module = pad_function(2**J)
# Create the filters
filters = filters_bank(self.M_padded, self.N_padded, J)
self.Psi = filters['psi']
self.Phi = [filters['phi'][j] for j in range(J)]