src/transformers/models/time_series_transformer/modeling_time_series_transformer.py/TimeSeriesTransformerForPrediction/output_distribution
class TimeSeriesTransformerForPrediction:    def output_distribution(self, params, loc=None, scale=None, trailing_n=None) -> torch.distributions.Distribution:
        sliced_params = params
        if trailing_n is not None:
            sliced_params = [p[:, -trailing_n:] for p in params]
        return self.distribution_output.distribution(sliced_params, loc=loc, scale=scale)