fft

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

Compute a fast Fourier transform.

Parameters:

• x (saiunit.Quantity | Array | ndarray | bool | number | bool | int | float | complex) – 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 (saiunit.Unit | None) – If provided, x is scaled by this unit before applying the function.

Returns:

result – The FFT result. Always unitless.

Return type:

jax.Array