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

Pytorch中的tensor维度理解

摘要

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

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

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

三维tensor

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

参考文献

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

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注