一个方括号一个维度,从外到内,维度从左到右
a=torch.tensor([[[1,2],[3,4]],
[[5,6],[7,8]]])
torch.Size([2, 2, 2])
内层的[1,2]为最后一个维度。
a=torch.tensor([[[1,2,2],[3,4,4]],
[[5,6,6],[7,8,8]]])
torch.Size([2, 2, 3])
再看并列的两个上一层的方括号,因为是两个,所以是中间的2.
a=torch.tensor([ [[1,2],[3,4]] ])
直接判断:
[1,2,2]