pytorch_forecasting.utils.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

pytorch_forecasting.utils.next_fast_len pytorch_forecasting.utils.profile