Pytorch - 添加2D张量元素的行

Question

我有以下张量：

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])

实际上，行数对应于变化的批量大小。

Answer 1

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

torch.sum(ts, dim=0)