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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。