_concatenate_output

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]]

Concatenate multiple batches of output dictionary.

Parameters:

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.

Returns:

concatenated output

Return type:

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