`torch_mist.critic.separable`

Module Contents

Classes

SeparableCritic

class torch_mist.critic.separable.SeparableCritic(f_x: torch.nn.Module | None = None, f_y: torch.nn.Module | None = None, temperature: float = 1.0)

Bases: torch_mist.critic.base.Critic

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