本文共 85 字，大约阅读时间需要 1 分钟。
pytorch.sum(a,dim=1)
官方文档：dim (int) – the dimension to reduce.
巧记：
按照dim指示的方向进行消掉(求和)
转载地址：http://hpfti.baihongyu.com/