Pytorch中的tensor维度理解

Pytorch中的tensor维度理解

摘要

面对pytorch编程中的tensor时，我不时会感到恐惧。对里面数据是怎么排布的，一直没有一个直观的理解。今天我想把这个事情弄清楚。本文从空间几何的角度给出了自己的理解。

打消心理恐惧，从三维学起

我们在编程中遇到的tensor大部分是三维、四维的，很少出现五个维度以及以上的。所以先弄懂三维，我们基本就够应付了。

三维tensor

第一维有三张图片。第二维度是两行，第三维度是三列。将这样一个三维的结构放在头脑中，能够帮助我们理解。

参考文献

[1] https://zhuanlan.zhihu.com/p/600971102