1. torch.sum(input, dim, out=None) 参数说明: input:输入的tensor矩阵. dim:求和的方向.若input为2维tensor矩阵,dim=0,对列求和:dim=1,对行求和.注意:输入的形状可为其他维度(3维或4维),可根据dim设定对相应的维度求和. out: 输出,一般不指定. 2. 例子 这里给了一个四维的tensor,代表的是输入到网络中的一张图片(batchSize,channel,height,width).dim=1,对通道求和. 2.…