mlx.core.fft.fftshift

fftshift(a: array, axes: Sequence[int] | None = None, stream: Stream | Device | None = None) → array

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

参数:

• a (array) -- The input array.

• axes (list(int), optional) -- Axes over which to perform the shift. If None, shift all axes.

返回:

The shifted array with the same shape as the input.

返回类型:

array