mlx.core.fft.fft

fft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array

一维离散傅里叶变换。

参数:

• a (array) – 输入数组。

• n (int, 可选) – 变换轴的大小。输入数组中对应的轴会被截断或补零以匹配 n。默认值为 a.shape[axis]。

• axis (int, 可选) – 执行 FFT 的轴。默认为 -1。

返回:

输入数组沿给定轴的 DFT。

返回类型:

array