在学习pytorch实现合并和求和运算时,经常会用到维度的概念,我在第一次接触时还是有点混乱的,在实践操作过程中,总结出一套自己的理解思路,故记录总结到博客上。
此处dim代表维度, 和numpy模块中的axis一样。
对于二维数组而言, dim=0代表行, dim=1代表列。
- 数组合并操作
- 数组求和操作
- dim=0代表行相加, 可以理解为所有行相加之和
- dim=1代表列相加,可以理解为所有列相加之和
在学习pytorch实现合并和求和运算时,经常会用到维度的概念,我在第一次接触时还是有点混乱的,在实践操作过程中,总结出一套自己的理解思路,故记录总结到博客上。
此处dim代表维度, 和numpy模块中的axis一样。
对于二维数组而言, dim=0代表行, dim=1代表列。