当前位置:主页 > 软件编程 > Python代码 >
时间:2021-03-27 09:29:49 | 栏目:Python代码 | 点击:次
不是concat的意思
import torch a = torch.ones([1,2]) b = torch.ones([1,2]) torch.stack([a,b],1) (0 ,.,.) = 1 1 1 1 [torch.FloatTensor of size 1x2x2]