jax.numpy.fft.rfft2
Warning
This page was created from a pull request (#9655).
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 ({"backward", "ortho", "forward"}, optional) –
- Returns
out – The result of the real 2-D FFT.
- Return type