mlx.core.fft.fft

目录

mlx.core.fft.fft#

fft(a: array, n: int | None = None, axis: int = -1, stream: Stream | Device | None = None) array#

One dimensional discrete Fourier Transform.

参数:
  • a (array) -- The input array.

  • n (int, optional) -- Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n. The default value is a.shape[axis].

  • axis (int, optional) -- Axis along which to perform the FFT. The default is -1.

返回:

The DFT of the input along the given axis.

返回类型:

array