Index of max in tensor

>>> a = torch.randn(4, 4)
>>> a
tensor([[ 1.3398,  0.2663, -0.2686,  0.2450],
        [-0.7401, -0.8805, -0.3402, -1.1936],
        [ 0.4907, -1.3948, -1.0691, -0.3132],
        [-1.6092,  0.5419, -0.2993,  0.3195]])
>>> torch.argmax(a)
tensor(0)

Source: pytorch.org

Similar codes

0 vote208 views1 answer
Preventing players from changing existing entries in tic tac toe game
Python
2 votes210 views3 answers
Condition python
Python
22 votes224 views6 answers
How to merge two dataframes
Python
0 vote211 views1 answer
Save impt
Python
0 vote227 views1 answer
Recorrer diccionario python
Python
157 votes203 views1 answer
Alphabet
Python
0 vote211 views1 answer
Python get user of pid
Python
0 vote225 views1 answer
Where to import render in django
Python
24 votes144 views4 answers
Valueerror: the truth value of a series is ambiguous. use a.empty, a.bool(), a.item(), a.any() or a.all().
Python
0 vote205 views1 answer
Set pixel pygame
Python