擅长:python、mysql、java
<p>我可以用下面的代码做到这一点。在</p>
<p>单矢量矩阵</p>
<pre><code>v = torch.arange(3)
M = torch.arange(8).view(2, 4)
# v: tensor([0, 1, 2])
# M: tensor([[0, 1, 2, 3],
# [4, 5, 6, 7]])
torch.mm(v.unsqueeze(1), M.view(1, 2*4)).view(3,2,4)
tensor([[[ 0, 0, 0, 0],
[ 0, 0, 0, 0]],
[[ 0, 1, 2, 3],
[ 4, 5, 6, 7]],
[[ 0, 2, 4, 6],
[ 8, 10, 12, 14]]])
</code></pre>
<p>对于一批向量和矩阵,可以很容易地使用<code>torch.bmm</code>进行扩展。在</p>
^{pr2}$