torch.stack

torch.stack函数是一个用于将多个张量在指定维度上进行堆叠的函数。它的作用是将多个相同形状的张量按照指定的维度进行连接，生成一个新的张量。与torch.cat函数不同的是，torch.stack会增加一个新的维度进行堆叠。[2]

举个例子来说明，假设我们有两个3x3的矩阵a和b，我们可以使用torch.stack函数将它们在不同的维度上进行堆叠。比如，使用dim=0，我们可以将a和b在第0维度上进行堆叠，生成一个2x3x3的张量。[1]

另外，我们也可以使用不同的维度进行堆叠。比如，使用dim=2，我们可以将a和b在第2维度上进行堆叠，生成一个3x3x2的张量。[3]

总结来说，torch.stack函数可以将多个张量在指定维度上进行堆叠，生成一个新的张量。它的作用类似于将多个矩阵按照指定维度进行拼接。