Pytorch中的tensor維度理解

摘要

面對pytorch編程中的tensor時，我不時會感到恐懼。對里面數據是怎么排布的，一直沒有一個直觀的理解。今天我想把這個事情弄清楚。本文從空間幾何的角度給出了自己的理解。

打消心理恐懼，從三維學起

我們在編程中遇到的tensor大部分是三維、四維的，很少出現五個維度以及以上的。所以先弄懂三維，我們基本就夠應付了。

三維tensor

第一維有三張圖片。第二維度是兩行，第三維度是三列。將這樣一個三維的結構放在頭腦中，能夠幫助我們理解。

參考文獻

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