CondAEModel.split_y_head_in CondAEModel.split_y_head_in(y: ndarray | Tensor) → Dict[str, ndarray | Tensor][source] Helper function to split the input of the model into the different heads.