Skip to content

DerivedWS

GlobalNVIDIAPyTorch

Import path: earth2studio.models.dx.DerivedWS

View source on GitHub View install commands

Documentation

Bases: Module

Calculates the Wind Speed (WS) magnitude from eastward and northward wind components for specified levels. The calculation is based on the formula: ws = sqrt(u^2 + v^2)

Parameters:

  • levels (list[int | str], default: [100] ) –

    Pressure / height levels to compute WS for. The resulting expected input fields are u and v wind components pairs for each level. E.g. for level 100 the input fields should be [u100, v100], by default [100]

__call__ 

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

Forward pass of diagnostic