jax.numpy.fft.fftfreq#

jax.numpy.fft.fftfreq(n,d=1.0,*,dtype=None,device=None)[source]#

Return sample frequencies for the discrete Fourier transform.

JAX implementation ofnumpy.fft.fftfreq(). Returns frequencies appropriatefor use with the outputs offft() andifft().

Parameters:

• n (int) – length of the FFT window

• d (ArrayLike) – optional scalar sample spacing (default: 1.0)

• dtype (DTypeLike |None) – optional dtype of returned frequencies. If not specified, JAX’s defaultfloating point dtype will be used.

• device (xla_client.Device |Sharding |None) – optionalDevice orShardingto which the created array will be committed.

Returns:

Array of sample frequencies, lengthn.

Return type:

Array

See also

• jax.numpy.fft.rfftfreq(): frequencies for use withrfft() andirfft().