torch的张量操作
一、torch.stacka = torch.tensor([[1,2,3]])b = torch.tensor([[4,5,6]])print(torch.stack((a,b),dim=0))print(torch.stack((a,b),dim=1))#输出:tensor([[[1, 2, 3]], [[4, 5, 6]]])tensor([[[1, 2, 3], [4, 5, 6]]])二、torch.cata = torch.
原创
2020-09-07 21:54:06 ·
1792 阅读 ·
0 评论