TypeError: fft_rfft() got an unexpected keyword argument 'onesided'

[turmeric-blend]

Hi, thanks for the open source code.

For pytorch 1.8.0, its required to use torch.fft.rfft() and torch.fft.irfft() instead of torch.rfft()/torch.irfft(). For the newer versions, there is no onesided argument. How do I update the code for fft to work for pytorch 1.8.0? Thanks.

[zotko]

Use the new torch.fft module functions, instead, by importing torch.fft and calling torch.fft.rfft() for one-sided output, or torch.fft.fft() for two-sided output (see link)

[Sudo42b]

Edit
ffted = torch.rfft(input, 1, onesided=False) to ffted = torch.view_as_real(torch.fft.fft(input, dim=1))
and
iffted = torch.irfft(time_step_as_inner, 1, onesided=False) to
iffted = torch.fft.irfft(torch.view_as_complex(time_step_as_inner), n=time_step_as_inner.shape[1], dim=1)