mlx.core.fft.fft2

Contents

mlx.core.fft.fft2#

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

Two dimensional discrete Fourier Transform.

Parameters:

  • a (array) – The input array.

  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

  • axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

Returns:

The DFT of the input along the given axes.

Return type:

array

Contents