CondAEModel.split_xy_out

CondAEModel.split_xy_out(x: ndarray | torch.Tensor, y: ndarray | torch.Tensor) → Tuple[Dict[str, ndarray | torch.Tensor], Dict[str, ndarray | torch.Tensor]]

Splits the output of the model according to the x and y heads.