Pytorch - 按元素添加二维张量的行
Pytorch - add rows of a 2D tensor element-wise
我有以下张量：
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])
实际上，行数对应于不同的批量大小。
您可以像这样对 axis/dimension 求和：
torch.sum(ts, dim=0)
我有以下张量：
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])
实际上，行数对应于不同的批量大小。
您可以像这样对 axis/dimension 求和：
torch.sum(ts, dim=0)