tensor.stack()

torch.stack(tensors, dim)

• tensors的size必须相同
• dim从0开始
• 特殊：dim为-1时，表示指定最后一个维度串联

关于维度：

a = torch.rand(3, 4)
    b = torch.rand(3, 4)
    c = torch.rand(3, 4)
    d = torch.stack((a, b, c), dim=-1)
    d1 = torch.stack((a, b, c), dim=1)
    d2 = torch.stack((a, b, c), dim=2)
    d3 = torch.stack((a, b, c), dim=0)
    print(f'a.size:{a.size()}')
    print(f'a:{a}')
    print(f'b.size:{b.size()}')
    print(f'b:{b}')
    print(f'c.size:{c.size()}')
    print(f'c:{c}')
    print(f'd.size:{d.size()}')
    print(f'd:{d}')
    print(f'd1.size:{d1.size()}')
    print(f'd1:{d1}')
    print(f'd2.size:{d2.size()}')
    print(f'd2:{d2}')
    print(f'd3.size:{d3.size()}')
    print(f'd3:{d3}')

输出结果：

a.size:torch.Size([3, 4])
a:tensor([[0.5166, 0.4962, 0.7092, 0.8800],
        [0.8129, 0.2019, 0.0429, 0.0798],
        [0.9910, 0.1685, 0.2235, 0.5326]])
b.size:torch.Size([3, 4])
b:tensor([[0.0963, 0.4543, 0.5801, 0.8981],
        [0.7436, 0.9575, 0.4813, 0.1266],
        [0.8621, 0.1629, 0.3639, 0.9070]])
c.size:torch.Size([3, 4])
c:tensor([[0.4439, 0.8803, 0.2745, 0.2690],
        [0.5564, 0.4358, 0.8667, 0.0335],
        [0.6743, 0.6172, 0.7626, 0.7893]])
d.size:torch.Size([3, 4, 3])
d:tensor([[[0.5166, 0.0963, 0.4439],
         [0.4962, 0.4543, 0.8803],
         [0.7092, 0.5801, 0.2745],
         [0.8800, 0.8981, 0.2690]],

        [[0.8129, 0.7436, 0.5564],
         [0.2019, 0.9575, 0.4358],
         [0.0429, 0.4813, 0.8667],
         [0.0798, 0.1266, 0.0335]],

        [[0.9910, 0.8621, 0.6743],
         [0.1685, 0.1629, 0.6172],
         [0.2235, 0.3639, 0.7626],
         [0.5326, 0.9070, 0.7893]]])
d1.size:torch.Size([3, 3, 4])
d1:tensor([[[0.5166, 0.4962, 0.7092, 0.8800],
         [0.0963, 0.4543, 0.5801, 0.8981],
         [0.4439, 0.8803, 0.2745, 0.2690]],

        [[0.8129, 0.2019, 0.0429, 0.0798],
         [0.7436, 0.9575, 0.4813, 0.1266],
         [0.5564, 0.4358, 0.8667, 0.0335]],

        [[0.9910, 0.1685, 0.2235, 0.5326],
         [0.8621, 0.1629, 0.3639, 0.9070],
         [0.6743, 0.6172, 0.7626, 0.7893]]])
d2.size:torch.Size([3, 4, 3])
d2:tensor([[[0.5166, 0.0963, 0.4439],
         [0.4962, 0.4543, 0.8803],
         [0.7092, 0.5801, 0.2745],
         [0.8800, 0.8981, 0.2690]],

        [[0.8129, 0.7436, 0.5564],
         [0.2019, 0.9575, 0.4358],
         [0.0429, 0.4813, 0.8667],
         [0.0798, 0.1266, 0.0335]],

        [[0.9910, 0.8621, 0.6743],
         [0.1685, 0.1629, 0.6172],
         [0.2235, 0.3639, 0.7626],
         [0.5326, 0.9070, 0.7893]]])
d3.size:torch.Size([3, 3, 4])
d3:tensor([[[0.5166, 0.4962, 0.7092, 0.8800],
         [0.8129, 0.2019, 0.0429, 0.0798],
         [0.9910, 0.1685, 0.2235, 0.5326]],

        [[0.0963, 0.4543, 0.5801, 0.8981],
         [0.7436, 0.9575, 0.4813, 0.1266],
         [0.8621, 0.1629, 0.3639, 0.9070]],

        [[0.4439, 0.8803, 0.2745, 0.2690],
         [0.5564, 0.4358, 0.8667, 0.0335],
         [0.6743, 0.6172, 0.7626, 0.7893]]])

Process finished with exit code 0

也即：
dim = -1 的情况：

d = [[  [a[0][0], b[0][0], c[0][0]],
        [a[0][1], b[0][1], c[0][1]],
        [a[0][2], b[0][2], c[0][2]],
        [a[0][3], b[0][3], c[0][3]]],

        [a[1][0], b[1][0], c[1][0]],
        [a[1][1], b[1][1], c[1][1]],
        [a[1][2], b[1][2], c[1][2]],
        [a[1][3], b[1][3], c[1][3]]],
          
        [a[2][0], b[2][0], c[2][0]],
        [a[2][1], b[2][1], c[2][1]],
        [a[2][2], b[2][2], c[2][2]],
        [a[2][3], b[2][3], c[2][3]]]

dim = 1 的情况：

d1 = [ [a[0],
        b[0],
        c[0]],

       [a[1], 
        b[1],
        c[1]],
  
       [a[2], 
        b[2], 
        c[2]]]

dim = 2 的情况：

d2同d

dim = 0 的情况：

d3 = [  a,
        b,
        c  ]