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 numpy.fft.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 ({"backward", "ortho", "forward"}, optional) –

Returns

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

Return type

ndarray