mlx.core.fft.fftshift

目录

mlx.core.fft.fftshift#

fftshift(a: array, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) array#

将零频率分量移到频谱中心。

参数:
  • a (array) – 输入数组。

  • axes (list(int), optional) – 执行移位的轴。如果为 None,则对所有轴执行移位。

返回:

移位后的数组,其形状与输入相同。

返回类型:

array