Pytorch中sum()函数的多维使用

函数参数：

torch.sum(input, dim, keepdim=False, *, dtype=None) → Tensor

当torch.sum(input tensor,dim)函数中出现多个维度的相加时如：

 

可以看出，pytorch执行sum函数的顺序是先将每一个1维度下的二维列表视为一个单位，所以每个二维列表里的三个一维度列表各为一个基本元素进行相加，得到[[6,6,6][15,15,15]]，再同样的思路将2维度下的int值视为一个基本元素进行相加， 得到最终[18,45]。在使用sum函数多维相加时，先确定单位，再确定基本元素就不会出错了。