运行别人的的源代码还报错，所以确定不是代码问题。

问了别人，应该是fft函数对应的torch版本问题，torch1.8.0版本之后的才是

torch.fft.fft2

根据网上的总结自己改的

#旧版          新版
torch.rfft   torch.fft.fft2
torch.irfft  torch.fft.ifft2

 还是报错了，函数中使用的参数定义应该也是不一样的