Dim1:维度为1的向量或者张量。
一般用在bias,一维的tensor。
dim=0的tensor是标量。
dim、size/shape中:2维指的是维数;size指整个的【2,2】,指tensor的具体形状。
四维的tensor适合图片,很适合卷积神经网络。
numel表示tensor所占内存大小。
Dim1:维度为1的向量或者张量。
一般用在bias,一维的tensor。
dim=0的tensor是标量。
dim、size/shape中:2维指的是维数;size指整个的【2,2】,指tensor的具体形状。
四维的tensor适合图片,很适合卷积神经网络。
numel表示tensor所占内存大小。