`torch_mist.estimators.utils`

Module Contents

Classes

FlippedMIEstimator

Functions

flip_estimator(→ torch_mist.estimators.MIEstimator)

class torch_mist.estimators.utils.FlippedMIEstimator(estimator: torch_mist.estimators.MIEstimator)

Bases: torch_mist.estimators.MIEstimator

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

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

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

torch_mist.estimators.utils.flip_estimator(estimator: torch_mist.estimators.MIEstimator) → torch_mist.estimators.MIEstimator