278bc3b0dfa36226d8818809e54f80edf14643a1,kymatio/scattering3d/backend/backend_skcuda.py,,complex_modulus,#Any#,15

Before Change

def complex_modulus(input_array):
    raise NotImplementedError(skcuda_notimplementederror)


def fft(input, inverse=False):
    

After Change

def complex_modulus(input_array):
    modulus = torch.zeros_like(input_array)
    modulus[..., 0] += torch.sqrt((input_array ** 2).sum(-1))
    return modulus


def fft(input, inverse=False):

Instances