Предположим, у меня есть следующий тензор, называемый 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))?
Спасибо,