Problem with torch.rfft

[Hramchenko]

Hello!

Thank you for a great work! I tried to run the code with pytorch 1.8 but got an error AttributeError: module 'torch' has no attribute 'rfft'. Could you tell me how to fix it?

Thanks.

[teboli]

Hi!

I think it is due to the fact that the newest version of Pytorch now has a dedicated package named torch.fft and that indeed contains rfft. You can either replace torch.rfft(x, 2, onesided=False) by torch.fft.fft2(x) or downgrade your pytorch version to 1.7 or an older version.

Let me know if it works.

[Hramchenko]

Thank you so much for your quick response.

Unfortunately, it didn't work. It seems that not only the package with fft has changed in pytorch, but also the signature of the function call :(