擅长:python、mysql、java
<p><strong>对于浮点张量</strong>,我用它来获取张量中元素的索引。</p>
<pre><code>print((torch.abs((torch.max(your_tensor).item()-your_tensor))<0.0001).nonzero())
</code></pre>
<p>在这里我想得到float张量中max_value的索引,你也可以像这样把你的值放进去,得到张量中任何元素的索引。</p>
<pre><code>print((torch.abs((YOUR_VALUE-your_tensor))<0.0001).nonzero())
</code></pre>