rfftn

Contents

rfftn#

class brainunit.fft.rfftn(a, s=None, axes=None, norm=None, **kwargs)#

Compute a multidimensional DFT of a real-valued array.

Unit-aware implementation of numpy.fft.rfftn(). The output unit is input_unit * second ** n_axes.

Parameters:

  • a (saiunit.Quantity | Array | ndarray | bool | number | bool | int | float | complex) – Real-valued input array.

  • s (Sequence[int] | None) – Effective shape of the input along axes.

  • axes (Sequence[int] | None) – Axes over which to compute the real DFT. None means all axes.

  • norm (str | None) – Normalization mode.

Returns:

N-dimensional real DFT of a.

Return type:

saiunit.Quantity | Array | ndarray | bool | number | bool | int | float | complex

See also

saiunit.fft.irfftn

Inverse of rfftn.

saiunit.fft.rfft

One-dimensional real DFT.

Examples

>>> import saiunit as u
>>> import saiunit.fft as sufft
>>> import jax.numpy as jnp
>>> x = jnp.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]) * u.meter
>>> X = sufft.rfftn(x)
Contents