函数torch.rfft()和torch.irfft()的问题

Question

我需要运行为旧版本的PyTorch编写的代码。我有1.9版。从版本1.8开始，PyTorch引入了torch.fft.rfft()和torch.fft.irfft()函数，它们的工作方式不同于旧的torch.rfft()和torch.irfft()。我不知道如何替换这些函数，以便使这段代码与旧版本的代码完全相同：

1.8版：

    fU = torch.rfft( u, 1, onesided=False)
    U = torch.irfft(fU, 1, onesided=False)
    torch.fft(x, x.ndim)

请帮帮我

Answer 1

正如您所提到的，torch.rfft()和torch.irfft()在输入和输出的格式上与torch.fft.rfft()和torch.fft.irfft()有差异。(见https://github.com/pytorch/pytorch/wiki/The-torch.fft-module-in-PyTorch-1.7)

在问题所在的GitHub页面PyTorch中，我对二维操作做了如下更改：

spectrum = torch.fft.rfft2(signal)
spectrum = torch.complex(spectrum[..., 0], spectrum[..., 1])
signal_recovered = torch.fft.irfft2(spectrum)