src/transformers/time_series_utils.py/DistributionOutput/_base_distribution
class DistributionOutput:    def _base_distribution(self, distr_args):
        if self.dim == 1:
            return self.distribution_class(*distr_args)
        else:
            return Independent(self.distribution_class(*distr_args), 1)