jax.lax.fft

jax.lax.fft(x, fft_type, fft_lengths)

Parameters

• fft_type (Union[FftType, str]) –

• fft_lengths (Sequence[int]) –