Skip to main content
`Ctrl`+`K`

# Array API Standard Support: `fft`

## Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function | torch | jax | dask
--- | --- | --- | ---
fft | ✔️ | ✔️ | ✔️
ifft | ✔️ | ✔️ | ✔️
fft2 | ✔️ | ✔️ | ✔️
ifft2 | ✔️ | ✔️ | ✔️
fftn | ✔️ | ✔️ | ✔️
ifftn | ✔️ | ✔️ | ✔️
rfft | ✔️ | ✔️ | ✔️
irfft | ✔️ | ✔️ | ✔️
rfft2 | ✔️ | ✔️ | ✔️
irfft2 | ✔️ | ✔️ | ✔️
rfftn | ✔️ | ✔️ | ✔️
irfftn | ✔️ | ✔️ | ✔️
hfft | ✔️ | ✔️ | ✔️
ihfft | ✔️ | ✔️ | ✔️
hfft2 | ✔️ | ✔️ | ✔️
ihfft2 | ✔️ | ✔️ | ✔️
hfftn | ✔️ | ✔️ | ✔️
ihfftn | ✔️ | ✔️ | ✔️
fftfreq | ✔️ | ✔️ | ✔️
rfftfreq | ✔️ | ✔️ | ✔️
fftshift | ✔️ | ✔️ | ✔️
ifftshift | ✔️ | ✔️ | ✔️
next_fast_len | N/A | N/A | N/A
prev_fast_len | N/A | N/A | N/A
dct | ✔️ | ✖ | ✔️
idct | ✔️ | ✔️ | ✔️
dst | ✔️ | ✖ | ✔️
idst | ✔️ | ✔️ | ✔️
dctn | ✔️ | ✔️ | ✔️
idctn | ✔️ | ✔️ | ✔️
dstn | ✔️ | ✔️ | ✔️
idstn | ✔️ | ✔️ | ✔️
fht | ✔️ | ✔️ | ✔️
ifht | ✔️ | ✔️ | ✔️
fhtoffset | N/A | N/A | N/A
set_backend | N/A | N/A | N/A
skip_backend | N/A | N/A | N/A
set_global_backend | N/A | N/A | N/A
register_backend | N/A | N/A | N/A
get_workers | N/A | N/A | N/A
set_workers | N/A | N/A | N/A

## Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function | cupy | torch | jax
--- | --- | --- | ---
fft | ✔️ | ✔️ | ✔️
ifft | ✔️ | ✔️ | ✔️
fft2 | ✔️ | ✔️ | ✔️
ifft2 | ✔️ | ✔️ | ✔️
fftn | ✔️ | ✔️ | ✔️
ifftn | ✔️ | ✔️ | ✔️
rfft | ✔️ | ✔️ | ✔️
irfft | ✔️ | ✔️ | ✔️
rfft2 | ✔️ | ✔️ | ✔️
irfft2 | ✔️ | ✔️ | ✔️
rfftn | ✔️ | ✔️ | ✔️
irfftn | ✔️ | ✔️ | ✔️
hfft | ✔️ | ✔️ | ✔️
ihfft | ✔️ | ✔️ | ✔️
hfft2 | ✔️ | ✔️ | ✔️
ihfft2 | ✔️ | ✔️ | ✔️
hfftn | ✔️ | ✔️ | ✔️
ihfftn | ✔️ | ✔️ | ✔️
fftfreq | ✔️ | ✔️ | ✔️
rfftfreq | ✔️ | ✔️ | ✔️
fftshift | ✔️ | ✔️ | ✔️
ifftshift | ✔️ | ✔️ | ✔️
next_fast_len | N/A | N/A | N/A
prev_fast_len | N/A | N/A | N/A
dct | ✖ | ✖ | ✖
idct | ✖ | ✖ | ✖
dst | ✖ | ✖ | ✖
idst | ✖ | ✖ | ✖
dctn | ✖ | ✖ | ✖
idctn | ✖ | ✖ | ✖
dstn | ✖ | ✖ | ✖
idstn | ✖ | ✖ | ✖
fht | ✔️ | ✔️ | ✔️
ifht | ✔️ | ✔️ | ✔️
fhtoffset | N/A | N/A | N/A
set_backend | N/A | N/A | N/A
skip_backend | N/A | N/A | N/A
set_global_backend | N/A | N/A | N/A
register_backend | N/A | N/A | N/A
get_workers | N/A | N/A | N/A
set_workers | N/A | N/A | N/A

## Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function | jax
--- | ---
fft | ✔️
ifft | ✔️
fft2 | ✔️
ifft2 | ✔️
fftn | ✔️
ifftn | ✔️
rfft | ✔️
irfft | ✔️
rfft2 | ✔️
irfft2 | ✔️
rfftn | ✔️
irfftn | ✔️
hfft | ✔️
ihfft | ✔️
hfft2 | ✔️
ihfft2 | ✔️
hfftn | ✔️
ihfftn | ✔️
fftfreq | ✔️
rfftfreq | ✔️
fftshift | ✔️
ifftshift | ✔️
next_fast_len | N/A
prev_fast_len | N/A
dct | ✖
idct | ✔️
dst | ✖
idst | ✔️
dctn | ✔️
idctn | ✔️
dstn | ✔️
idstn | ✔️
fht | ✔️
ifht | ✔️
fhtoffset | N/A
set_backend | N/A
skip_backend | N/A
set_global_backend | N/A
register_backend | N/A
get_workers | N/A
set_workers | N/A