torch_mist.estimators.transformed.implementations.binned

Module Contents

Classes

BinnedMIEstimator

class torch_mist.estimators.transformed.implementations.binned.BinnedMIEstimator(quantize_x: torch_mist.quantization.functions.QuantizationFunction | None = None, quantize_y: torch_mist.quantization.functions.QuantizationFunction | None = None, temperature: float = 1.0)

Bases: torch_mist.estimators.transformed.base.TransformedMIEstimator

lower_bound = True