标签: python pytorch tensor
假设我有一个名为a的以下张量:
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))?
谢谢