CondAEModel.split_xy_in

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

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