mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
bb6565ef14
* add fftshift and ifftshift fft helpers * address comments * axes have to be iterable * fix fp error in roll + add test --------- Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local>
25 lines
211 B
ReStructuredText
25 lines
211 B
ReStructuredText
.. _fft:
|
|
|
|
FFT
|
|
===
|
|
|
|
.. currentmodule:: mlx.core.fft
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
fft
|
|
ifft
|
|
fft2
|
|
ifft2
|
|
fftn
|
|
ifftn
|
|
rfft
|
|
irfft
|
|
rfft2
|
|
irfft2
|
|
rfftn
|
|
irfftn
|
|
fftshift
|
|
ifftshift
|