`torch_mist.estimators.generative.implementations.ba`

Module Contents

Classes

class torch_mist.estimators.generative.implementations.ba.BA(q_Y_given_X: pyro.distributions.ConditionalDistribution, entropy_y: torch.Tensor | None = None)

Bases: torch_mist.estimators.generative.base.ConditionalGenerativeMIEstimator

lower_bound: bool = True

infomax_gradient: Dict[str, bool]

mutual_information(x: torch.Tensor, y: torch.Tensor) → torch.Tensor