Pytorch函数笔记--stack

a = torch.tensor([[1,2,3],
                  [4,5,6],
                  [7,8,9]])
b = torch.tensor([[10,20,30],
                  [40,50,60],
                  [70,80,90]])
c = torch.tensor([[100,200,300],
                  [400,500,600],
                  [700,800,900]])
d =torch.stack((a,b,c),dim =1)

输出：