numpy
有两种方式拼接矩阵:
#1. 使用concatenate级联操作
np.concatenate((a,b),axis=0)
#2. 使用np.vstack沿矩阵行进行拼接,np.hstack沿矩阵列拼接
np.vstack((a,b))
pytorch
有两种方式拼接张量:
#1. 使用concatenate级联操作
torch.cat((a,b), dim=0)
#2.使用stack,dim控制不同tensor维度
torch.stack([a,b], dim=0)
参考
[1] PyTorch 常用方法总结
[2] numpy实现矩阵的拼接