earth2studio.models.dx
.SolarRadiationAFNO6H#
- class earth2studio.models.dx.SolarRadiationAFNO6H(core_model, freq, era5_mean, era5_std, ssrd_mean, ssrd_std, orography, landsea_mask, sincos_latlon)[source]#
Solar Radiation AFNO diagnostic model for 6-hour frequency predictions. Predicts the accumulated global surface solar radiation over 6 hours [Jm^-2].
- Parameters:
core_model (Module)
freq (str)
era5_mean (Tensor)
era5_std (Tensor)
ssrd_mean (Tensor)
ssrd_std (Tensor)
orography (Tensor)
landsea_mask (Tensor)
sincos_latlon (Tensor)
- __call__(x, coords)#
Forward pass of diagnostic
- Parameters:
x (Tensor)
coords (OrderedDict[str, ndarray])
- Return type:
tuple[Tensor, OrderedDict[str, ndarray]]