mlx.core.fft.fftshift

Contents

mlx.core.fft.fftshift#

fftshift(a: array, axes: int | Sequence[int] | None = None, stream: Stream | mlx.core.ThreadLocalStream | Device | None = None) → array#

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

Parameters:

  • a (array) – The input array.

  • axes (int or list(int), optional) – Axis or axes over which to perform the shift. If None, shift all axes.

Returns:

The shifted array with the same shape as the input.

Return type:

array

Contents