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.

参数:

• 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.

返回:

The shifted array with the same shape as the input.

返回类型:

array