perturbation.Gaussian

Import path: earth2studio.perturbation.Gaussian

View source on GitHub

Documentation

Standard Gaussian peturbation

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.

• seed (int | None, default: None ) –

  Seed for the internal generator, by default None.

__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

Returns:

• tuple[torch.Tensor, CoordSystem]: –

  Output tensor and respective coordinate system dictionary

Examples using earth2studio.perturbation.Gaussian

Model Hook Injection: PerturbationAdding model noise by using custom hooks. IO Backend PerformanceLeverage different IO backends for storing inference results.