jax.numpy.fft.fftshift

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

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

LAX-backend implementation of numpy.fft.fftshift().

Original docstring below.

This function swaps half-spaces for all axes listed (defaults to all). Note that y[0] is the Nyquist component only if len(x) is even.

Parameters:

• x (array_like) – Input array.

• axes (int or shape tuple, optional) – Axes over which to shift. Default is None, which shifts all axes.

Returns:

y – The shifted array.

Return type:

ndarray