`torch_mist.estimators.transformed.implementations.flip`

Module Contents

Classes

FlippedMIEstimator

class torch_mist.estimators.transformed.implementations.flip.FlippedMIEstimator(base_estimator: torch_mist.estimators.MIEstimator)

Bases: torch_mist.estimators.TransformedMIEstimator

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

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

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

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

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