mlx_rs::fft

Function irfft_device

Source 
pub fn irfft_device(
    a: impl AsRef<Array>,
    n: impl Into<Option<i32>>,
    axis: impl Into<Option<i32>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

The inverse of rfft().

The output has the same shape as the input except along axis in which case it has size n.

ยงParams

  • a: The input array.
  • n: Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n // 2 + 1. The default value is a.shape[axis] // 2 + 1.
  • axis: Axis along which to perform the FFT. The default is -1.