torch_mist.distributions.transforms.base

Module Contents

Classes

DistributionModule	Helper class that provides a standard way to create an ABC using
ConditionalDistributionModule	Helper class that provides a standard way to create an ABC using
ConditionalTransformedDistributionModule
TransformedDistributionModule	Helper class that provides a standard way to create an ABC using

class torch_mist.distributions.transforms.base.DistributionModule(validate_args: bool = False)

Bases: torch.distributions.Distribution, torch.nn.Module, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

__repr__()

Return repr(self).

class torch_mist.distributions.transforms.base.ConditionalDistributionModule

Bases: pyro.distributions.ConditionalDistribution, torch.nn.Module, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

class torch_mist.distributions.transforms.base.ConditionalTransformedDistributionModule(base_dist: pyro.distributions.ConditionalDistribution | torch.distributions.Distribution, transforms: pyro.distributions.ConditionalTransform | List[pyro.distributions.ConditionalTransform] | Dict[str, pyro.distributions.ConditionalTransform | torch.distributions.Transform] | torch.distributions.Transform | List[torch.distributions.Transform] | None, cached: bool = True)

Bases: pyro.distributions.ConditionalTransformedDistribution, torch.nn.Module

condition(context)

clear_cache()

__repr__()

class torch_mist.distributions.transforms.base.TransformedDistributionModule(base_dist: torch.distributions.Distribution, transforms: torch.distributions.Transform | List[torch.distributions.Transform] | Dict[str, torch.distributions.Transform] | None, cached: bool = True)

Bases: DistributionModule

Helper class that provides a standard way to create an ABC using inheritance.

rsample(sample_shape=torch.Size())

log_prob(value)

__repr__()

Return repr(self).