`torch_mist.estimators.hybrid.implementations.resampled`

Module Contents

Classes

ResampledHybridMIEstimator

class torch_mist.estimators.hybrid.implementations.resampled.ResampledHybridMIEstimator(generative_estimator: torch_mist.estimators.generative.base.GenerativeMIEstimator, discriminative_estimator: torch_mist.estimators.discriminative.base.DiscriminativeMIEstimator)

Bases: torch_mist.estimators.hybrid.base.HybridMIEstimator

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