Priors

class jolideco.priors.Priors(modules: Mapping[str, Module] | None = None)

Bases: ModuleDict

Dict of mutiple priors

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

__call__(fluxes)

Evaluate all priors

Methods Documentation

__call__(fluxes)

Evaluate all priors

Parameters:

fluxes (tuple of ~torch.Tensor) – Tuple of flux tensors

Returns:

log_prior – Log prior value

Return type:

~torch.tensor