擅长:python、mysql、java
<p>在将最后一个轴推到前面之后,我们可以在3D数组<code>np.linalg.inv</code>上使用<code>a</code>。然后,我们可以利用<code>einsum</code>作为最终输出的矢量化方式-</p>
<pre><code>p = np.linalg.inv(a.transpose(2,0,1))
c = np.einsum('ijk,kli->ij',p,b)[...,None]
</code></pre>
<p>获得最终输出<code>c</code>的一个更简单的方法是使用<code>np.matmul/@-operator</code>-</p>
<pre><code>c = p@b.transpose(2,0,1)
</code></pre>
<p>因此,整个过程可以转换为一个线性-</p>
<pre><code>c = np.linalg.inv(a.transpose(2,0,1))@b.transpose(2,0,1)
</code></pre>