_concatenate_output#

pytorch_forecasting.models.base_model._concatenate_output(output: List[Dict[str, List[Union[List[Tensor], Tensor, bool, int, str, ndarray]]]]) Dict[str, Union[Tensor, ndarray, List[Union[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]]]]