_concatenate_output#

pytorch_forecasting.models.base._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.

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