fft2

Contents

fft2#

class saiunit.fft.fft2(a, s=None, axes=(-2, -1), norm=None, **kwargs)#

Compute a two-dimensional discrete Fourier transform along given axes.

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

Parameters:

  • a (saiunit.Quantity | Array | ndarray | bool | number | bool | int | float | complex) – Input array with a.ndim >= 2.

  • s (Sequence[int] | None) – Shape (length-2) of the output along each axes entry.

  • axes (Sequence[int]) – Axes over which to compute the 2-D DFT.

  • norm (str | None) – Normalization mode.

Returns:

Two-dimensional DFT of a.

Return type:

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

See also

saiunit.fft.ifft2

Two-dimensional inverse DFT.

saiunit.fft.fftn

N-dimensional 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]]) * u.meter
>>> X = sufft.fft2(x)
>>> x_back = sufft.ifft2(X)
Contents