Helper Functions

The following functions are auxiliary (helper) functions frequently used in radiative transfer problems.

compile(model, filename)[source]

Compile a PyTorch model to a TorchScript file.

Parameters:

  • model – The PyTorch model to compile.

  • filename – The name of the file to save the compiled model.

Returns:

A compiled version of the model.

bbflux_wavenumber(*args, **kwargs)

Overloaded function.

  1. bbflux_wavenumber(wave: torch.Tensor, temp: typing.SupportsFloat, ncol: typing.SupportsInt = 1) -> torch.Tensor

  2. bbflux_wavenumber(wn1: typing.SupportsFloat, wn2: typing.SupportsFloat, temp: torch.Tensor = 1) -> torch.Tensor

  3. bbflux_wavenumber(wn1: torch.Tensor, wn2: torch.Tensor, temp: torch.Tensor) -> torch.Tensor

bbflux_wavelength(wave: torch.Tensor, temp: SupportsFloat, ncol: SupportsInt = 1) → torch.Tensor
calc_dz_hypsometric(pres: torch.Tensor, temp: torch.Tensor, g_ov_R: torch.Tensor) → torch.Tensor
interpn(query: collections.abc.Sequence[torch.Tensor], coords: collections.abc.Sequence[torch.Tensor], lookup: torch.Tensor, extrapolate: bool = False) → torch.Tensor