jax.numpy.fft.fftshift#

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

Shift zero-frequency fft component to the center of the spectrum.

JAX implementation ofnumpy.fft.fftshift().

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.ifftshift(): inverse offftshift.

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