brainpy.math.fft.rfft2

brainpy.math.fft.rfft2(a, s=None, axes=(-2, -1), norm=None)

Similar to jax.numpy.fft.rfft2 function, while it is compatible with brainpy Array/Variable.

Note that this function is also compatible with:

1. NumPy or PyTorch syntax when receiving out argument.

2. PyTorch syntax when receiving keepdim or dim argument.

3. TensorFlow syntax when receiving keep_dims argument.