jax.numpy.fft.rfft2ΒΆ

jax.numpy.fft.rfft2(a, s=None, axes=(-2, -1), norm=None)[source]ΒΆ

Compute the 2-dimensional FFT of a real array.

LAX-backend implementation of rfft2(). Original docstring below.

Parameters
  • a (array) – Input array, taken to be real.

  • s (sequence of ints, optional) – Shape of the FFT.

  • axes (sequence of ints, optional) – Axes over which to compute the FFT.

  • norm ({None, "ortho"}, optional) –

    New in version 1.10.0.

Returns

out – The result of the real 2-D FFT.

Return type

ndarray

See also

rfftn()

Compute the N-dimensional discrete Fourier Transform for real input.

Notes

This is really just rfftn with different default behavior. For more details see rfftn.