Tensor Dimensions

1.number of tensor’s dimension

python 
import torch 
t=torch.ones(1,2,3,4) #创建dim=4,size=[1,2,3,4]的tensor
t #显示t
t.size() #显示t的size

通过tensor t的显示可以看出，tensor的dim=从外到遇到第一个元素的 ‘[‘ 个数

2.index of tensor’s dimension

t.flatten(start_dim=0).size()
t.flatten(start_dim=1).size()
t.flatten(start_dim=2).size()
t.flatten(start_dim=3).size()
t.flatten(start_dim=-1).size()

通过把t从各个dim展平判断出tensor t的dim下标依次是[0,1,2,3],另外dim 3==-1。

note

关于torch.flatten()此method介绍见：https://pytorch.org/docs/stable/generated/torch.flatten.html#torch.flatten