pytorch_forecasting.models.base_model._concatenate_output(output: List[Dict[str, List[List[Tensor] | Tensor | bool | int | str | ndarray]]]) Dict[str, Tensor | ndarray | List[Tensor | int | bool | str]][source]#

Concatenate multiple batches of output dictionary.


output (List[Dict[str, List[Union[List[torch.Tensor], torch.Tensor, bool, int, str, np.ndarray]]]]) – list of outputs to concatenate. Each entry corresponds to a batch.


concatenated output

Return type:

Dict[str, Union[torch.Tensor, np.ndarray, List[Union[torch.Tensor, int, bool, str]]]]