torch_mist.estimators.transformed.implementations.pq

Module Contents

Classes

PQ

class torch_mist.estimators.transformed.implementations.pq.PQ(q_QY_given_X: pyro.distributions.ConditionalDistribution, quantize_y: torch_mist.quantization.functions.QuantizationFunction, temperature: float = 1.0)

Bases: torch_mist.estimators.transformed.base.TransformedMIEstimator