PyTorch torch.stack 函数：将张量序列堆叠到新维度

torch.stack 是一个 PyTorch 函数，用于将张量序列沿着新的维度进行堆叠，生成一个新的张量。新维度的尺寸等于输入张量序列的数量。例如，如果输入张量序列的形状为 (3, 4, 5)，则堆叠后的张量形状为 (3, 3, 4, 5)，其中第二个维度的尺寸为 3，表示有 3 个输入张量。

torch.stack 函数的语法如下：

torch.stack(sequence, dim=0, out=None)

其中：

• sequence 是一个张量序列。
• dim 是要堆叠的新维度的索引（默认为 0）。
• out 是可选的输出张量。如果提供了 out，则结果将被写入该张量，而不是创建一个新的张量。

torch.stack 函数的返回值是一个新的张量，其中包含输入张量序列沿着新维度堆叠后的结果。