fft

Contents

fft#

class saiunit.lax.fft(x, fft_type, fft_lengths, unit_to_scale=None, **kwargs)#

Compute a fast Fourier transform.

Parameters:
  • x (Union[saiunit.Quantity, Array, ndarray, number, bool]) – Input array. Must be dimensionless or scaled via unit_to_scale.

  • fft_type (FftType | str) – The type of FFT (e.g., 'FFT', 'IFFT', 'RFFT', 'IRFFT').

  • fft_lengths (Sequence[int]) – The lengths of each FFT dimension.

  • unit_to_scale (Optional[saiunit.Unit]) – If provided, x is scaled by this unit before applying the function.

Returns:

result – The FFT result. Always unitless.

Return type:

Array