トーチは軸に沿ってテンソルを合計します

Question

ipdb> outputs.size() 
torch.Size([10, 100]) 
ipdb> print sum(outputs,0).size(),sum(outputs,1).size(),sum(outputs,2).size() 
(100L,) (100L,) (100L,) 

どのようにして列を合計しますか？

Answer 1

最も簡単で最も良い解決策は、torch.sum()を使用することです。すべての列（すなわち、行ごとに）上で要約する

torch.sum(outputs, dim=0) # size = [1, ncol] 

：すべての行（すなわち、各列の）上で要約する

torch.sum(outputs) # gives back a scalar 

：テンソルのすべての要素を合計する

。

torch.sum(outputs, dim=1) # size = [nrow, 1]