causica.distributions.distribution_module¶

Module Contents¶

Classes¶

Attributes¶

causica.distributions.distribution_module.DistributionType_co[source]¶

class causica.distributions.distribution_module.DistributionModule(*args, **kwargs)[source]¶

Bases: Generic[DistributionType_co], torch.nn.Module

Baseclass for modules returning distributions.

Useful e.g. to create variational approximations of distributions.

Subclasses are expected to implement a forward method that returns a concrete td.Distribution and should usually inherit from a conrete version of this class, i.e. DistributionModule[<td.Distribution subclass>].

__call__(*args, **kwargs) → DistributionType_co[source]¶

Return a td.Distribution.