jax.numpy.fft.fftfreq

jax.numpy.fft.fftfreq(n, d=1.0, *, dtype=None, device=None)

返回离散傅里叶变换的采样频率。

JAX 实现 numpy.fft.fftfreq()。返回适合与 fft() 和 ifft() 的输出一起使用的频率。

参数：

• n (int) – FFT 窗口的长度

• d (ArrayLike) – 可选的标量样本间距（默认值：1.0）

• dtype (DTypeLike | None | None) – 返回频率的可选数据类型。如果未指定，则将使用 JAX 的默认浮点数据类型。

• device (xla_client.Device | 分片 | None | None) – 创建的数组将提交到的可选 Device 或 Sharding。

返回:

样本频率数组，长度为 n。

返回类型:

数组

另请参阅

• jax.numpy.fft.rfftfreq(): 用于 rfft() 和 irfft() 的频率。