71. torch.stack

[CHH3213]

函数作用： 函数stack()对序列数据内部的张量进行扩维拼接，指定维度由程序员选择、大小是生成后数据的维度区间。

• 示例

  
  
  T1 = torch.tensor([[1, 2, 3],
               [4, 5, 6],
               [7, 8, 9]])
  
  T2 = torch.tensor([[10, 20, 30],
               [40, 50, 60],
               [70, 80, 90]])

print(torch.stack((T1,T2),dim=0).shape) print(torch.stack((T1,T2),dim=1).shape) print(torch.stack((T1,T2),dim=2).shape)

输出为：

torch.Size([2, 3, 3]) torch.Size([3, 2, 3]) torch.Size([3, 3, 2])