老版本pytorch实现多维度傅里叶(Fourier Transform)变换
在新版本pytorch中可以轻易实现多维度的傅里叶变换:
x = torch.rand(10, 10, dtype=torch.complex64)
fftn = torch.fft.fftn(x)
老版本中没有torch.fft.fftn()函数,因此需要进行函数嵌套实现
x = torch.rand(10, 10, dtype=torch.complex64)
two_ffts = torch.fft(torch.fft(x, dim=0), dim=1)