擅长:python、mysql、java
<p>您可以使用<a href="https://pytorch.org/docs/stable/torch.html#torch.einsum" rel="noreferrer">einsum</a></p>
<pre><code>torch.einsum('bp, bqr->bpqr', v, M) #batch version (v.shape=(b,p) M.shape=(b,q,r))
torch.einsum('p, qr->pqr', v, M) #non batch version
</code></pre>