def _params_from_output(
        self, model_output: torch.Tensor
    ) -> Union[tuple[torch.Tensor, ...], torch.Tensor]:
        """
        Returns the distribution parameters, obtained from the raw model outputs
        (e.g. applies softplus or sigmoids to get parameters in the expected domains).
        """

        Returns the distribution parameters, obtained from the raw model outputs
        (e.g. applies softplus or sigmoids to get parameters in the expected domains).
        
darts/utils/likelihood_models/torch.py