fftn

Computes the n-dimensional discrete Fourier transform.

Abstract Signature:

fftn(x: array, s: Optional[Sequence[int]], axes: Optional[Sequence[int]], norm: Literal[backward, ortho, forward])

PyTorch

API: torch.fft.fftn
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.fft.fftn
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.fft.fftn
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.fft.fftn
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.fft.fftn
Strategy: Direct Mapping
Official Docs ↗