jax.numpy.fft.fftshiftΒΆ

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

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

LAX-backend implementation of 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