AttributeError：模块“torch.fft”没有属性“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: 模块 'torch.fft' 没有属性 '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 是在PyTorch version 1.8.0 中引入的。您需要升级到此版本或更高版本才能使用它。