填充堆叠#

pytorch_forecasting.utils._utils.padded_stack(tensors: List[Tensor], side: str = 'right', mode: str = 'constant', value: int | float = 0) Tensor[来源]#

沿着第一维堆叠张量,并在最后一维中填充以确保它们的大小相等。

Parameters:
  • 张量 (列表[torch.Tensor]) – 要堆叠的张量列表

  • side (str) – 填充的边 - “left” 或 “right”。默认为 “right”。

  • mode (str) – ‘constant’, ‘reflect’, ‘replicate’ 或 ‘circular’. 默认:‘constant’

  • value (Union[int, float]) – 用于常量填充的值

Returns:

堆叠张量

Return type:

torch.Tensor