擅长:python、mysql、java
<p>从这个问题中很难理解你想要的结果是什么,但我认为你所追求的结果可以很容易地得到,就像这样:</p>
<pre><code>y = (X.T * W[:,dim2,dim3]).sum(axis=0)
</code></pre>
<p>比较正确性和速度:</p>
^{pr2}$
<p>再快一点就可以用了</p>
<pre><code>y = np.einsum('ij,ji->i', X, W[:,dim2,dim3])
</code></pre>