擅长:python、mysql、java
<p>我不完全理解您的<code>dim2</code>和<code>dim3</code>数组是什么,以及<code>mask</code>是如何构造的,但是根据您的描述,您需要与此类似的内容:</p>
<pre><code>ni = 10
a, b, c = 43, 91, 67
X = np.random.rand(ni*a*b, c)
W = np.random.rand(c, a, b)
X = X.reshape(ni, a*b, c)
W = W.reshape(c, a*b)
y = np.einsum('ijk, kj -> ij', X, W)
y = y.reshape(-1)
</code></pre>
<p>如果您使用工作代码更新问题,即<code>dim2</code>和<code>dim3</code>的完整描述,我们可以对其进行微调以返回完全相同的结果(如果还没有)。在</p>