Pytorch - 逐元素添加二维张量行

【问题标题】：Pytorch - add rows of a 2D tensor element-wisePytorch - 逐元素添加二维张量行
【发布时间】：2019-04-03 17:00:57
【问题描述】：

我有以下张量：

ts = torch.tensor([[1,2,3],[4,6,7],[8,9,10]])
> tensor([[ 1,  2,  3],
    [ 4,  6,  7],
    [ 8,  9, 10]])

我正在寻找一个 pytorch 通用操作，它可以像这样按元素添加所有行：

ts2 = ts[0]+ts[1]+ts[2]
print(ts2)
> tensor([13, 17, 20])

实际上，行数对应于不同的批大小。

【问题讨论】：

标签： add pytorch

【解决方案1】：

您可以像这样对轴/维度求和：

torch.sum(ts, dim=0)

【讨论】：