jax.lax.fft

内容

jax.lax.fft#

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

  • fft_lengths (Sequence[int])