pytorch-在元素上添加二维张量的行

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)