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#
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 ofaalongaxes.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: