评价此页

torch.fft.rfftfreq#

torch.fft.rfftfreq(n, d=1.0, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) Tensor#

计算具有信号大小 nrfft() 的采样频率。

注意

rfft() 返回厄米单边输出,因此只返回正频率项。对于长度为 n 的实数 FFT,并且输入的间隔单位为 d,频率为

f = torch.arange((n + 1) // 2) / (d * n)

注意

对于偶数长度,奈奎斯特频率 f[n/2] 可以视为负数或正数。与 fftfreq() 不同,rfftfreq() 始终将其视为正数。

参数
  • n (int) – 实数 FFT 的长度

  • d (float, optional) – 采样长度尺度。FFT 输入的单个样本之间的间隔。默认假定单位间隔,除以实际间隔得到物理频率单位的结果。

关键字参数
  • out (Tensor, optional) – 输出张量。

  • dtype (torch.dtype, optional) – 返回张量的期望数据类型。默认值:如果 None,则使用全局默认值(参见 torch.set_default_dtype())。

  • layout (torch.layout, optional) – 返回张量的期望布局。默认值:torch.strided

  • device (torch.device, optional) – 返回张量的期望设备。默认值:如果 None,则使用当前设备作为默认张量类型(参见 torch.set_default_device())。对于 CPU 张量类型,device 是 CPU;对于 CUDA 张量类型,device 是当前 CUDA 设备。

  • requires_grad (bool, optional) – 是否应自动记录返回张量上的操作。默认值:False

示例

>>> torch.fft.rfftfreq(5)
tensor([0.0000, 0.2000, 0.4000])
>>> torch.fft.rfftfreq(4)
tensor([0.0000, 0.2500, 0.5000])

fftfreq() 的输出相比,我们发现 f[2] 处的奈奎斯特频率改变了符号:>>> torch.fft.fftfreq(4) tensor([ 0.0000, 0.2500, -0.5000, -0.2500])