I have the same issue with torch 1.8.1, while it all works fine with torch 1.7.1 torch.fft was depreciated in this version of torch: pytorch/pytorch#42175 Annoyingly, when you try to use new versions ...