src/transformers/time_series_utils.py/DistributionOutput/__init__
class DistributionOutput:    def __init__(self, dim: int = 1) -> None:
        self.dim = dim
        self.args_dim = {k: dim * self.args_dim[k] for k in self.args_dim}