mlx_rs::fft

Function fftn_device

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

n-dimensional discrete Fourier Transform.

ยงParams

  • a: The input array.
  • s: 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 if not specified.
  • axes: Axes along which to perform the FFT. The default is None in which case the FFT is over the last len(s) axes are or all axes if s is also None.