pytorch:tensor的合并与分割

import torch

cat

• dim后面接的数字表示从哪个维度上进行拼接，注意：其他维度的size必须要两两相等

a = torch.rand(30,6)#a记录了班上前30个同学的6门考试成绩
b = torch.rand(25,6)#b记录了班上后30个同学的6门考试成绩
torch.cat([a,b],dim=0).shape 

torch.Size([55, 6])

a = torch.rand(5,3) #5个同学，a只记录了4门课的成绩
b = torch.rand(5,2)#还是这5个同学，b记录了另外2门课的