`torch_mist.estimators.generative.implementations.club`

Module Contents

Classes

class torch_mist.estimators.generative.implementations.club.CLUB(q_Y_given_X: pyro.distributions.ConditionalDistribution, neg_samples: int = 0)

Bases: torch_mist.estimators.generative.implementations.l1out.L1Out

infomax_gradient: Dict[str, bool]

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