jax.lax.fft

Contents

jax.lax.fft#

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

  • fft_lengths (Sequence[int]) –