mlx_rs::fft

Function fft2_device

Source 
pub fn fft2_device<'a>(
    a: impl AsRef<Array>,
    s: impl IntoOption<&'a [i32]>,
    axes: impl IntoOption<&'a [i32]>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

Two dimensional discrete Fourier Transform.

ยงParams

  • a: The input array.
  • s: Size of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match s. The default value is the sizes of a along axes.
  • axes: Axes along which to perform the FFT. The default is [-2, -1].