cupy.fft.rfft2

cupy.fft.rfft2(a, s=None, axes=(-2, -1), norm=None)[source]

Compute the two-dimensional FFT for real input.

Parameters:
  • a (cupy.ndarray) – Array to be transform.
  • s (None or tuple of ints) – Shape to use from the input. If s is not given, the lengths of the input along the axes specified by axes are used.
  • axes (tuple of ints) – Axes over which to compute the FFT.
  • norm (None or "ortho") – Keyword to specify the normalization mode.
Returns:

The transformed array which shape is specified by s and type will convert to complex if the input is other. The length of the last axis transformed will be s[-1]//2+1.

Return type:

cupy.ndarray