AttributeError: module 'torch.fft' has no attribute 'fftfreq'

Question

我按照 https://pytorch.org/docs/stable/generated/torch.fft.fftshift.html#torch.fft.fftshift

import torch.fft
f = torch.fft.fftfreq(4)
a = torch.fft.fftshift(f)
print(a)

得到错误

AttributeError: module 'torch.fft' has no attribute 'fftfreq'

我尝试了 pip torch==1.7.0+cu110 和 pip torch==1.7.1+cu110 以及 conda pytorch==1.7.1 和 cudatoolkit=11.0。 其他人也有同样的问题https://discuss.pytorch.org/t/unable-to-use-correctly-the-new-torch-fft-module/104560/6 但是换成torch1.7.0并没有解决问题

如何正确使用torch.fft？

Answer 1

函数torch.fft.fftfreq was introduced in PyTorch version 1.8.0。您需要升级到此版本或更高版本才能使用它。