numpy 和pytorch的矩阵/张量拼接

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实现矩阵的拼接