torch.stack()

官方解释：沿着一个新维度对输入张量序列进行连接。 序列中所有的张量都应该为相同形状。

浅显说法：把多个2维的张量凑成一个3维的张量；多个3维的凑成一个4维的张量…以此类推，也就是在增加新的维度进行堆叠。

import torch
a = torch.tensor([[1, 2, 3],
     [4, 5, 6],
      [7, 8, 9]])
b = torch.tensor([[10, 20, 30],
        [40, 50, 60],
        [70, 80, 90]])
print(a.shape,b.shape)

a和b都是2维的，多个2维的可以凑成一个3维的张量，dim设置为2和axis设置为-1效果是一样的。