a是形状为[2,3]的torch.tensor
b是形状为[2,3]的torch.tensor
期望输出:c.shape[2,2,3]
代码如下
a = a.unsqueeze(0)
b = b.unsqueeze(0)
c = torch.cat((a, b),0)
ps.初学时期踩坑记录:网上查的先将a,b转换成list再转换成nparray再转换成torch.tensor的方法属于是又麻烦错误又多,建议不要使用。
a是形状为[2,3]的torch.tensor
b是形状为[2,3]的torch.tensor
期望输出:c.shape[2,2,3]
代码如下
a = a.unsqueeze(0)
b = b.unsqueeze(0)
c = torch.cat((a, b),0)
ps.初学时期踩坑记录:网上查的先将a,b转换成list再转换成nparray再转换成torch.tensor的方法属于是又麻烦错误又多,建议不要使用。