火炬中的列总和
Column sum in torch
我如何沿着 torch 中的列求和?我有一个128*1024的张量,我想通过对所有行求和得到一个1*1024的张量。
例如:
一个:
1 2 3
4 5 6
我要b
5 7 9
为此,您可以使用 sum 方法。
torch.sum(a,1)
一般来说,您可以指定要求和的任何轴。
torch.sum(a,axis)
(要对行求和,可以使用 axis=2)
我如何沿着 torch 中的列求和?我有一个128*1024的张量,我想通过对所有行求和得到一个1*1024的张量。
例如: 一个:
1 2 3
4 5 6
我要b
5 7 9
为此,您可以使用 sum 方法。
torch.sum(a,1)
一般来说,您可以指定要求和的任何轴。
torch.sum(a,axis)
(要对行求和,可以使用 axis=2)