CondAEModel.split_x_head_out

CondAEModel.split_x_head_out(x: ndarray | torch.Tensor) → Dict[str, ndarray | torch.Tensor]

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