torch.srft

`torch.srft` 是 PyTorch（一个开源机器学习库）中用于快速傅立叶变换 (FFT) 和逆快傅立叶变换 (IFFT) 的函数。它实现了单循环移位快速傅立叶变换 (Single-Round Fast Fourier Transform)，这通常比标准的 FFT 更节省内存，因为它只需要一次通过数据。这个函数在处理大规模的数据时特别有用，尤其是在有限自动机（Finite State Automata）的应用、语音信号处理或其他需要高效频域操作的场景。

`srft` 函数接收两个参数：
1. 输入张量 `input`：输入的复数或实数值序列。
2. 通道维度 `dim`：表示输入数据中的频率分量所在的轴。

示例用法：

import torch

# 创建一个二维输入张量表示声音信号
signal = torch.randn(1000, 2)

# 进行快速傅立叶变换
fft_output = torch.srft(signal, dim=1)