CondVAEModel.split_y_head_out

CondVAEModel.split_y_head_out(y: ndarray | torch.Tensor) Dict[str, ndarray | torch.Tensor][source]

Helper function to split the output of the model into the different heads.