擅长:python、mysql、java
<p>您可以使用<a href="https://pytorch.org/docs/stable/torch.html#torch.topk" rel="nofollow noreferrer">topk</a>函数</p>
<p>例如:</p>
<pre><code>import torch
t = torch.tensor([5.7, 1.4, 9.5, 1.6, 6.1, 4.3])
values,indices = t.topk(2)
print(values)
print(indices)
</code></pre>
<p>结果是:</p>
<pre><code>tensor([9.5000, 6.1000])
tensor([2, 4])
</code></pre>