pytorch中实现FFT变换

在PyTorch中，可以使用`torch.fft`模块实现FFT变换。下面是一个简单的例子：

import torch

# 输入信号
x = torch.tensor([1.0, 2.0, 1.0, -1.0, 1.5, 0.5])

# 执行FFT变换
fft_x = torch.fft.fft(x)

# 输出FFT结果
print(fft_x)

在上面的例子中，我们首先定义了一个张量`x`作为输入信号，然后使用`torch.fft.fft`函数执行FFT变换，得到了变换后的结果`fft_x`。最后，我们将结果打印出来。

需要注意的是，`torch.fft.fft`函数默认会对最后一个维度进行FFT变换。如果输入信号是一维的，那么可以直接使用上面的代码进行FFT变换。如果输入信号是多维的，那么需要指定要进行FFT变换的维度。例如，假设输入信号是一个形状为`(batch_size, channel, height, width)`的四维张量，我们想要对每个通道分别进行FFT变换，可以使用以下代码：

import torch

# 输入信号
x = torch.randn(4, 3, 32, 32)

# 对每个通道分别进行FFT变换
fft_x = torch.fft.fft(x, dim=[2, 3])

# 输出FFT结果
print(fft_x.shape)

在上面的例子中，我们使用`dim=[2, 3]`参数指定要对第三维和第四维进行FFT变换，得到的结果是一个形状为`(4, 3, 32, 32)`的张量，表示每个通道的FFT变换结果。