pytorch维度的理解
在学习pytorch实现合并和求和运算时，经常会用到维度的概念，我在第一次接触时还是有点混乱的，在实践操作过程中，总结出一套自己的理解思路，故记录总结到博客上。此处dim代表维度， 和numpy模块中的axis一样。对于二维数组而言， dim=0代表行， dim=1代表列。数组合并操作数组求和操作dim=0代表行相加， 可以理解为所有行相加之和dim=1代表列相加，可以理解为所有列相加之和...
·
在学习pytorch实现合并和求和运算时，经常会用到维度的概念，我在第一次接触时还是有点混乱的，在实践操作过程中，总结出一套自己的理解思路，故记录总结到博客上。
此处dim代表维度， 和numpy模块中的axis一样。
对于二维数组而言， dim=0代表行， dim=1代表列。
- 数组合并操作
- 数组求和操作
- dim=0代表行相加， 可以理解为所有行相加之和
- dim=1代表列相加，可以理解为所有列相加之和
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
2
0- 0
已为社区贡献5条内容
所有评论(0)