sum

x.dim(dim=0, keepdim=True)

0是按列，1是按行

keepdim是保留行和列这两个维度

【推广】 免费学中医，健康全家人
原文地址：https://www.cnblogs.com/liujianing/p/13083521.html