CondVAEModel.encode

CondVAEModel.encode(x: torch.Tensor) → Dict[str, torch.Tensor]

Encode the input data into a latent representation.

Parameters:

x (torch.Tensor) – The input data.

Returns:

Dict[str, torch.Tensor] – A dictionary containing the latent representation ‘z’ and the conditional part ‘y’.