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