src/transformers/time_series_utils.py/NegativeBinomialOutput/_base_distribution
class NegativeBinomialOutput:    def _base_distribution(self, distr_args) -> Distribution:
        total_count, logits = distr_args
        if self.dim == 1:
            return self.distribution_class(total_count=total_count, logits=logits)
        else:
            return Independent(self.distribution_class(total_count=total_count, logits=logits), 1)