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) – 返回频率的可选数据类型。如果未指定，将使用 JAX 默认的浮点数据类型。

• device (xla_client.Device | Sharding | None) – 可选的 Device 或 Sharding，创建的数组将被提交至该设备或分片策略。

返回:

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

返回类型:

Array

另请参阅

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