版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/83031466
import torch a= torch.FloatTensor([1,0,1]) b=torch.FloatTensor([0,0,1]) c=torch.cat((a.unsqueeze(0),b.unsqueeze(0)),0) print(c)
import torch a= torch.FloatTensor([1,0,1]) b=torch.FloatTensor([0,0,1]) c=torch.cat((a.unsqueeze(0),b.unsqueeze(0)),0) print(c)