【深度学习】PyTorch tensor张量维度理解

参考: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]
请添加图片描述

猜你喜欢

转载自blog.csdn.net/guai7guai11/article/details/124808142