padded_stack¶
- pytorch_forecasting.utils.padded_stack(tensors: List[torch.Tensor], side: str = 'right', mode: str = 'constant', value: Union[int, float] = 0) torch.Tensor [source]¶
Stack tensors along first dimension and pad them along last dimension to ensure their size is equal.
- Parameters
tensors (List[torch.Tensor]) – list of tensors to stack
side (str) – side on which to pad - “left” or “right”. Defaults to “right”.
mode (str) – ‘constant’, ‘reflect’, ‘replicate’ or ‘circular’. Default: ‘constant’
value (Union[int, float]) – value to use for constant padding
- Returns
stacked tensor
- Return type
torch.Tensor