jax.numpy.fft.fftfreq

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

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

numpy.fft.fftfreq() 的 JAX 实现。返回适用于 fft() 和 ifft() 输出的频率。

参数：

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

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

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

• device (xla_client.Device | Sharding | None | None) – 可选的 Device 或 Sharding，用于提交创建的数组。

返回：

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

返回类型：

数组

另请参阅

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