`torch_mist.estimators.generative.implementations.gm`

Module Contents

Classes

class torch_mist.estimators.generative.implementations.gm.GM(q_XY: torch_mist.distributions.joint.base.JointDistribution, q_Y: torch.distributions.Distribution | torch_mist.distributions.joint.base.JointDistribution, q_X: torch.distributions.Distribution | torch_mist.distributions.joint.base.JointDistribution)

Bases: torch_mist.estimators.generative.base.JointGenerativeMIEstimator

property q_X

property q_Y

batch_loss(x: torch.Tensor, y: torch.Tensor) → torch.Tensor