擅长:python、mysql、java
<p>最简单和最好的解决方案是使用<code>torch.sum()</code>。</p>
<p>要对张量的所有元素求和:</p>
<pre><code>torch.sum(outputs) # gives back a scalar
</code></pre>
<p>对所有行(即每列)求和:</p>
<pre><code>torch.sum(outputs, dim=0) # size = [1, ncol]
</code></pre>
<p>对所有列(即每行)求和:</p>
<pre><code>torch.sum(outputs, dim=1) # size = [nrow, 1]
</code></pre>