jax.numpy.fft.ifftshift#

jax.numpy.fft.ifftshift(x,axes=None)[source]#

The inverse ofjax.numpy.fft.fftshift().

JAX implementation ofnumpy.fft.ifftshift().

Parameters:

• x (ArrayLike) – N-dimensional array array of frequencies.

• axes (None |int |Sequence[int]) – optional integer or sequence of integers specifying which axes toshift. If None (default), then shift all axes.

Returns:

A shifted copy ofx.

Return type:

Array

See also

• jax.numpy.fft.fftshift(): inverse ofifftshift.

• jax.numpy.fft.fftfreq(): generate FFT frequencies.

Examples

Generate FFT frequencies withfftfreq():

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

Usefftshift() to shift the zero-frequency entryto the middle of the array:

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

Unshift withifftshift to recover the original frequencies:

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