数据类型
Type check
Dimension 0 / rank 0
维度Dimension:0 / 秩rank:0
NumPy中的维度(dimension)、轴(axis)、秩(rank)的含义
Dim 0
Dim 1 / rank 1
Linear Input(线性输入)
Dim 1
Dim 2
Linear input batch(线性批量输入)
Dim 3
RNN Input Batch(循环神经网络批量输入)
Dim 4
CNN: [b, c, h, w]
卷积神经网络
Mixed
a.numel() # 占内存大小
a.dim() # 返回长度