`earth2studio.models.dx`.Identity#

class earth2studio.models.dx.Identity[source]#

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

__call__(x, coords)[source]#

Forward pass of diagnostic

Parameters:

x (Tensor)
coords (OrderedDict[str, ndarray])

Return type:

tuple[Tensor, OrderedDict[str, ndarray]]