torch.sum()用法

标签：input   lis   维度   数据   ima   none   lse   列表   http   

torch.sum()对输入的tensor数据的某一维度求和，一共两种用法
１．torch.sum(input, dtype=None)
２．torch.sum(input, list: dim, bool: keepdim=False, dtype=None) → Tensor
　
input:输入一个tensor
dim:要求和的维度，可以是一个列表
keepdim:求和之后这个dim的元素个数为１，所以要被去掉，如果要保留这个维度，则应当keepdim=True

torch.sum()用法

标签：input   lis   维度   数据   ima   none   lse   列表   http   

原文地址：https://www.cnblogs.com/wmc258/p/14544241.html