perturbation.Brown

Import path: earth2studio.perturbation.Brown

View source on GitHub

Documentation

Lat/Lon 2D brown noise

Parameters:

• noise_amplitude (float | Tensor, default: 0.05 ) –

  Noise amplitude, by default 0.05. If a tensor, this must be broadcastable with the input data.

• reddening (int, default: 2 ) –

  Reddening in Fourier space, by default 2

__call__

__call__(
    x: Tensor, coords: CoordSystem
) -> tuple[Tensor, CoordSystem]

Apply perturbation method

Parameters:

• x (Tensor) –

  Input tensor intended to apply perturbation on

• coords (CoordSystem) –

  Ordered dict representing coordinate system that describes the tensor, must contain "lat" and "lon" coordinates in last two dims

Returns:

• tuple[torch.Tensor, CoordSystem]: –

  Output tensor and respective coordinate system dictionary