`torch_mist.data.utils`

Module Contents

Functions

`prepare_samples`(→ Union[torch.Tensor, Dict[str, ...)
`compute_mi`(→ float)

torch_mist.data.utils.prepare_samples(distribution: torch_mist.distributions.JointDistribution, n_samples: int) → torch.Tensor | Dict[str, torch.Tensor]

torch_mist.data.utils.compute_mi(distribution: torch_mist.distributions.JointDistribution, **kwargs) → float