在pytorch1.4.0版本中torch.irfft(torch.rfft(a, 1), 1, signal_sizes=(a.shape[-1],))转换成pytorch1.8.0的要怎么写
时间: 2023-03-28 13:03:16 浏览: 62
在pytorch1.8.版本中,可以使用torch.fft.irfft(torch.fft.rfft(a, dim=1), dim=1, n=a.shape[-1])来替换torch.irfft(torch.rfft(a, 1), 1, signal_sizes=(a.shape[-1],))。
相关问题
torch.rfft和torch.irfft
torch.rfft和torch.irfft是PyTorch中用于实现实数输入的快速傅里叶变换(FFT)和逆傅里叶变换(IFFT)的函数。在旧版的PyTorch中,可以使用torch.rfft函数进行实数输入的FFT操作,并返回一个复数输出。而在新版的PyTorch中,可以使用torch.fft.rfft函数进行实数输入的FFT操作,并返回一个复数输出。另外,在新版中,为了获得“one-side output”,可以使用torch.fft.rfft函数。而如果想要获得“two-side output”,则可以使用torch.fft.fft函数。
pytorch1.12.1 中有torch.nn.functional包吗
PyTorch 1.12.1 版本中也有`torch.nn.functional`包,可以使用它来调用各种神经网络函数。你可以在代码中这样导入它:
```python
import torch.nn.functional as F
```
然后,你就可以使用`F`作为前缀来调用`torch.nn.functional`包中的函数,例如:
```python
x = torch.randn(2, 3)
y = F.relu(x)
```