`Identity`¶

GlobalNVIDIAPyTorch

Import path: earth2studio.models.dx.Identity

Documentation¶

Bases: Module

Identity diagnostic that is coordinate insensitive. Typically used for testing.

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

Forward pass of diagnostic