Skip to main content
Ctrl+K

jax.lax.fft#

jax.lax.fft(x, fft_type, fft_lengths)[source]#
Parameters:
  • fft_type (Union[FftType, str]) –

  • fft_lengths (Sequence[int]) –

previous

jax.lax.expm1

next

jax.lax.floor