擅长:python、mysql、java
<p>您正在寻找<a href="https://pytorch.org/docs/stable/tensors.html#torch.Tensor.index_add_" rel="nofollow noreferrer">^{<cd1>}</a></p>
<pre><code>A.index_add_(0, B, C)
</code></pre>
<p>注意,<code>B</code>应该是<code>torch.long</code>(它是一个索引向量),而{<cd4>}应该是{<cd5>}类型,与{<cd6>}的类型相同。<br/>
此外,如果<code>A</code>和{<cd4>}是多维张量,则可以使用第一个<code>dim</code>参数沿不同维度进行求和。在</p>