生成数据
生成一个矩阵维度为(2,3)的矩阵:
b=torch([ [0,1,2],
[3,4,5] ])
那么我们从b这个数据来分析如得到维度为(2,3)
分析维度
首先,最外侧有两个中括号,即torch([ [ ,可以很容易得到这个数据b共有两维。
关键来了:
我们根据每一层中括号【】对数据进行拆分(剥洋葱):
第一维,[ [0,1,2],
[3,4,5] ]
由于最外层中括号数为2,则里面只有一个中括号(2-1)的元素有2个,分别为[0,1,2]和[3,4,5],所以第一维度为2
第二维,只看上一维元素中的一个,如[0,1,2], 由于此时最外层中括号数为1,则里面没有中括号(1-1)的元素为3,分别为0,1,2,所以第二维度为3
再举一例加深理解
同理可得c,d,f 的维度。再那维度为(2,1,3)的d举例子吧。
d=torch([[[0,1,2]],
[[3,4,5]]])
很容易知道共有三维。
第一维:[[[0,1,2]],
[[3,4,5]]],可以看出,最外层(第三层中括号)里包含了两个中括号(3-1)的元素有两个,分别是[[0,1,2]], [[3,4,5]] 所以第一维度为2
第二维:只看上一维的一个元素,如**[** [0,1,2] ],最外层(第二层中括号)里包含了一个中括号(2-1)元素有1个,为[0,1,2],所以第二维度为1
第三维:只看上一维的一个元素,[ 0,1,2 ],最外层(第一层中括号)里包含了没有中括号(1-1)的元素有3个,分别为0,1,2,所以第三维度为3
所以d的总的维度为(2,1,3)