torch.stack重复堆叠

torch.stack函数用于将多个tensor在指定维度上进行堆叠，生成一个新的tensor。它将这些tensor按顺序堆叠在一起，生成的新tensor的维度会在指定维度上加一。

具体来说，torch.stack函数有三个参数：tensors是要堆叠的tensor组成的列表，dim是指定的堆叠维度，out是可选参数，用于指定输出的tensor。

下面是一个示例：

import torch

x = torch.tensor([[1, 2, 3], [4, 5, 6]])
m = torch.tensor([[1, 1, 1], [2, 2, 2]])

result = torch.stack([x, m], dim=0)
print(result)

输出结果为：

tensor([[[1, 2, 3],
         [4, 5, 6]],

        [[1, 1, 1],
         [2, 2, 2]]])

在这个示例中，我们将x和m两个tensor在第0维（即行）进行堆叠，生成了一个新的tensor。