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)