CondVAEModel.split_x_head_in

CondVAEModel.split_x_head_in(x: ndarray | torch.Tensor) Dict[str, ndarray | torch.Tensor][source]

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