擅长:python、mysql、java
<p>我想是我自己弄到的,不过还是会分享解决方案:</p>
<p>因为只得到矩阵乘法的对角线</p>
<pre><code>> Z = N.diag(X.dot(Y))
</code></pre>
<p>相当于X行和Y列的标量乘积之和,上一条语句相当于:</p>
<pre><code>> Z = (X * Y.T).sum(-1)
</code></pre>
<p>对于原始变量,这意味着:</p>
<pre><code>> result = (A.dot(B) * A).sum(-1)
</code></pre>
<p>如果我错了,请纠正我,但这应该是。。。</p>