Pytorch中的tensor维度理解
文章目录
- Pytorch中的tensor维度理解
- 摘要
- 打消心理恐惧,从三维学起
- 三维tensor
- 参考文献
摘要
面对pytorch编程中的tensor时,我不时会感到恐惧。对里面数据是怎么排布的,一直没有一个直观的理解。今天我想把这个事情弄清楚。本文从空间几何的角度给出了自己的理解。
打消心理恐惧,从三维学起
我们在编程中遇到的tensor大部分是三维、四维的,很少出现五个维度以及以上的。所以先弄懂三维,我们基本就够应付了。
三维tensor
第一维有三张图片。第二维度是两行,第三维度是三列。将这样一个三维的结构放在头脑中,能够帮助我们理解。
参考文献
[1] https://zhuanlan.zhihu.com/p/600971102