TensorFlow NUFFT DocumentationΒΆ

TensorFlow NUFFT is a fast, native non-uniform fast Fourier transform op for TensorFlow. It provides:

  • Fast CPU/GPU kernels. The TensorFlow framework automatically handles device placement as usual.

  • A simple, well-documented Python interface.

  • Gradient definitions for automatic differentiation.

  • Shape functions to support static shape inference.

The underlying algorithm is based on the NUFFT implementation by the Flatiron Institute. Please refer to FINUFFT and cuFINUFFT for more details.