saiunit.fft module

Unit-aware FFT (Fast Fourier Transform) functions.

Provides wrappers for jax.numpy.fft that correctly transform units between time/space and frequency domains.

Functions that Changing Unit

fft	Compute a one-dimensional discrete Fourier transform along a given axis.
rfft	Compute a one-dimensional DFT of a real-valued array.
ifft	Compute a one-dimensional inverse discrete Fourier transform.
irfft	Compute a real-valued one-dimensional inverse DFT.
fft2	Compute a two-dimensional discrete Fourier transform along given axes.
fftn	Compute a multidimensional discrete Fourier transform.
rfft2	Compute a two-dimensional DFT of a real-valued array.
rfftn	Compute a multidimensional DFT of a real-valued array.
ifft2	Compute a two-dimensional inverse discrete Fourier transform.
ifftn	Compute a multidimensional inverse discrete Fourier transform.
irfft2	Compute a real-valued two-dimensional inverse DFT.
irfftn	Compute a real-valued multidimensional inverse DFT.
fftfreq	Return sample frequencies for the discrete Fourier transform.
rfftfreq	Return sample frequencies for the real discrete Fourier transform.

Functions that Keeping Unit

fftshift	Shift zero-frequency FFT component to the center of the spectrum.
ifftshift	Inverse of saiunit.fft.fftshift().