torch_mist.estimators.multi

Module Contents

Classes

MultiMIEstimator

class torch_mist.estimators.multi.MultiMIEstimator(estimators: Dict[Tuple[str, str], torch_mist.estimators.base.MIEstimator])

Bases: torch_mist.estimators.base.MIEstimator

broadcast_function(function_name: str, **variables) → Dict[Tuple[str, str], torch.Tensor]

loss(**variables) → torch.Tensor

batch_loss(**variables) → torch.Tensor

mutual_information(**variables) → Dict[Tuple[str, str], torch.Tensor]

log_ratio(**variables) → Dict[Tuple[str, str], torch.Tensor]

forward(**variables) → torch.Tensor