Just a suggestion on Fourier transformation APIs for users whose pytorch version >= 1.8.
nxznm opened this issue · comments
This line could be changed to return torch.fft.irfftn(torch.conj(torch.fft.rfftn(a, (-1))) * torch.fft.rfftn(b, (-1)), (-1)) if your pytorch version >= 1.8.