如何找到出现最大值的火炬张量索引?

时间:2019-10-30 18:33:22

标签: python pytorch tensor

假设我有一个名为a的以下张量:

>>> a = torch.randn(10)
Out[32]: 
tensor([-0.6936,  1.2862,  0.2096, -0.8629, -3.0095, -0.7957, -0.9336,  0.0622,
        -0.3510, -0.1518])

如何找到出现最大值的该张量的索引(在这种情况下,位置1(= 1.2862))?

谢谢

0 个答案:

没有答案