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