擅长:python、mysql、java
<p>替换:</p>
<pre><code>error[:, np.argmax(label, axis=1)] -=1
</code></pre>
<p>使用:</p>
<pre><code>error[np.arange(error.shape[0]), np.argmax(label, axis=1)] -=1
</code></pre>
<p>当然了</p>
<pre><code>loss = error[np.arange(error.shape[0]), np.argmax(label, axis=1)].sum()
</code></pre>
<p>在您的示例中,您正在更改和求和<code>error[0,3]</code>和<code>error[1,1]</code>,简而言之<code>error[[0,1],[3,1]]</code>。你知道吗</p>