参考:pytorch 中维度(Dimension)概念的理解
以3×2×3的tensor为例
tensor([[[0.0932, 0.7748, 0.1215],
[0.6337, 0.3522, 0.1062]],
[[0.5779, 0.6993, 0.0230],
[0.9226, 0.5299, 0.1861]],
[[0.0227, 0.3016, 0.9368],
[0.7684, 0.7787, 0.6171]]])
torch.Size([3, 2, 3])
维度0,加法,即把每个二维数据按照对应位置,相加即可,且其输出的形状为[2,3]
维度1,加法,即对每个二维矩阵,行不变,按列相加,然后组合在一块,且其输出的形状为[3,3]
维度2,加法,即对每个二维矩阵,列不变,按行相加,然后组合在一块,且其输出的形状为[3,2]