torch_mist.estimators.discriminative.implementations.dummy

Module Contents

Classes

DummyDiscriminativeMIEstimator

class torch_mist.estimators.discriminative.implementations.dummy.DummyDiscriminativeMIEstimator(neg_samples: int = 1)

Bases: torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator

_approx_log_partition(x: torch.Tensor, y: torch.Tensor, f_: torch.Tensor, log_w: torch.Tensor | None) → torch.Tensor