CondAEModel.decode

CondAEModel.decode(y: torch.Tensor, z: array | torch.Tensor) → torch.Tensor[source]

Decode the latent representation into the original data space.

Parameters:

y (torch.Tensor) – The conditional data.
z (Union[np.array, torch.Tensor]) – The latent representation to decode.

Returns:

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