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) –

Returns

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

Return type

ndarray