`torch_mist.estimators.hybrid.implementations.hybrid_pq`

Module Contents

Classes

PQHybridMIEstimator

class torch_mist.estimators.hybrid.implementations.hybrid_pq.PQHybridMIEstimator(q_QY_given_X: pyro.distributions.ConditionalDistribution, quantize_y: torch_mist.quantization.QuantizationFunction, discriminative_estimator: torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator, temperature: float = 1.0)

Bases: torch_mist.estimators.hybrid.base.HybridMIEstimator

property quantize_y: Callable[[torch.Tensor], torch.LongTensor]

disable_batch_validation()

enable_batch_validation()

sample_negatives(x: torch.Tensor, y: torch.Tensor) → Tuple[torch.Tensor, torch.Tensor, torch.Tensor | None]