CondVAEModel.decode

CondVAEModel.decode(y: ndarray | torch.Tensor, z: ndarray | torch.Tensor | None = None) → torch.Tensor[source]

Decode the latent representation into the original data space.

Parameters:

y (torch.Tensor) – The conditional data.
z (Union[np.ndarray, torch.Tensor]) – The latent representation to decode. If None, a latent representation is sampled from a normal distribution.

Returns:

torch.Tensor – A tensor containing the reconstructed (generated) data.