CondVAEModel.split_xy_out

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

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