mlx.core.fft.fftn

目录

mlx.core.fft.fftn#

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

n维离散傅里叶变换。

参数:

  • a (array) – 输入数组。

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

  • axes (list(int), optional) – 执行 FFT 的轴。默认值为 None，在这种情况下，FFT 将作用于最后的 len(s) 个轴，如果 s 也为 None，则作用于所有轴。

返回值:

输入数组沿给定轴的 DFT。

返回类型:

array

目录