pytorch_forecasting.models.base_model._concatenate_output

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

base_model pytorch_forecasting.models.base_model._torch_cat_na