mlx.core.fft.fft2

fft2(a: array, s: Optional[Sequence[int]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array

二维离散傅里叶变换。

参数：

• a (array) – 输入数组。

• s (list(int), 可选) – 变换轴的尺寸。输入中相应的轴会被截断或填充零以匹配 s 中的尺寸。默认值是 a 沿 axes 的尺寸。

• axes (list(int), 可选) – 执行 FFT 的轴。默认值为 [-2, -1]。

返回值：

输入沿给定轴的 DFT。

返回类型：

array