Skip to main content

rfftfreq

Function rfftfreq 

Source
pub fn rfftfreq(n: usize, d: f64) -> Result<Array>
Expand description

Returns the nonnegative discrete Fourier transform sample frequencies for a real transform.

The transform length is n and the sample spacing is d.

ยงExample

use mlx_rs::{fft::rfftfreq, Device, Dtype};

Device::set_default(&Device::cpu());
let frequencies = rfftfreq(4, 0.5).unwrap();
assert_eq!(frequencies.shape(), &[3]);
assert_eq!(frequencies.dtype(), Dtype::Float32);