jax.numpy.fft.irfft2

Contents

jax.numpy.fft.irfft2#

jax.numpy.fft.irfft2(a, s=None, axes=(-2, -1), norm=None)[source]#

Computes the inverse of rfft2.

LAX-backend implementation of numpy.fft.irfft2().

Original docstring below.

Parameters:
  • a (array_like) – The input array

  • s (sequence of ints, optional) – Shape of the real output to the inverse FFT.

  • axes (sequence of ints, optional) – The axes over which to compute the inverse fft. Default is the last two axes.

  • norm ({"backward", "ortho", "forward"}, optional)

Returns:

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

Return type:

ndarray