mlx_rs::fft

Function irfft2

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

The inverse of rfft2().

Note the input is generally complex. The dimensions of the input specified in axes are padded or truncated to match the sizes from s. The last axis in axes is treated as the real axis and will have size s[s.len()-1] // 2 + 1.

ยง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 except for the last axis which has size s[s.len()-1] // 2 + 1. The default value is the sizes of a along axes.
  • axes: Axes along which to perform the FFT. The default is [-2, -1].