91eda80f9a542ce027ae77e861c9107182070b2d,kymatio/scattering3d/backend/torch_backend.py,,fft,#Any#Any#,96

Before Change

    if not _is_complex(input):
        raise TypeError("The input should be complex (e.g. last dimension is 2)")
    if inverse:
        return torch.ifft(input, 3)
    return torch.fft(input, 3)


def cdgmm3d(A, B, inplace=False):

After Change

        raise TypeError("The input should be complex (e.g. last dimension is 2)")
    if inverse:
        return torch.view_as_real(torch.fft.ifftn(torch.view_as_complex(input), dim=[-1, -2, -3]))
    return torch.view_as_real(torch.fft.fftn(torch.view_as_complex(input), dim=[-1, -2, -3]))


def cdgmm3d(A, B, inplace=False):
    Complex pointwise multiplication.

Instances