jax.numpy.fft.ifftshift

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

jax.numpy.fft.fftshift() 的逆运算。

JAX 对 numpy.fft.ifftshift() 的实现。

参数:

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

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

返回:

已移位的 x 的副本。

返回类型:

Array

另请参阅

• jax.numpy.fft.fftshift()：ifftshift 的逆运算。

• 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)