jax.numpy.fft.fftshift

jax.numpy.fft.fftshift(x, axes=None)

将零频率 fft 分量移到频谱中心。

numpy.fft.fftshift() 的 JAX 实现。

参数:

• x (ArrayLike) – 频率的 N 维数组。

• axes (None | int | Sequence[int] | None) – 可选的整数或整数序列，指定要移动的轴。 如果为 None（默认），则移动所有轴。

返回:

x 的移动副本。

返回类型:

数组

另请参阅

• jax.numpy.fft.ifftshift(): fftshift 的逆。

• jax.numpy.fft.fftfreq(): 生成 FFT 频率。

示例

使用 fftfreq() 生成 FFT 频率。

>>> freq = jnp.fft.fftfreq(5)
>>> freq
Array([ 0. ,  0.2,  0.4, -0.4, -0.2], dtype=float32)

使用 fftshift 将零频率条目移动到数组的中间。

>>> shifted_freq = jnp.fft.fftshift(freq)
>>> shifted_freq
Array([-0.4, -0.2,  0. ,  0.2,  0.4], dtype=float32)

使用 ifftshift() 取消移动，以恢复原始频率。

>>> jnp.fft.ifftshift(shifted_freq)
Array([ 0. ,  0.2,  0.4, -0.4, -0.2], dtype=float32)