jax.numpy.fft.fftshift

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

將零頻率 FFT 組件移動到頻譜中心。

JAX 實作的 numpy.fft.fftshift()。

參數:

• x (ArrayLike) – N 維陣列，表示頻率。

• axes (None | int | Sequence[int] | None) – 選用的整數或整數序列，指定要移動的軸。如果為 None (預設值)，則移動所有軸。

傳回:

x 的移動後副本。

傳回類型:

Array

另請參閱

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