`torch_mist.distributions.transforms.implementations.normalize`

Module Contents

Classes

class torch_mist.distributions.transforms.implementations.normalize.EMANormalize(input_dim: int, epsilon: float = 1e-06, gamma: float = 0.99, normalize_inverse=True)

Bases: torch_mist.distributions.transforms.implementations.linear.Linear

_update_params(t: torch.Tensor)

_inverse(y: torch.Tensor) → torch.Tensor

_call(x: torch.Tensor) → torch.Tensor