填充堆叠#
- 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