CondVAEModel.split_x_head_out CondVAEModel.split_x_head_out(x: ndarray | Tensor) → Dict[str, ndarray | Tensor][source] Helper function to split the output of the model into the different heads.