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 vote211 views1 answer
Unbreakable box made of diamond
Python
1 vote175 views1 answer
Minute range python
Python
0 vote183 views1 answer
Discord.File(fp=image_binary,filename=name) discord py
Python
1 vote193 views2 answers
Django class based views
Python Django
5 votes217 views1 answer
Add bearer token in python request
Python
0 vote194 views1 answer
How to sort a list in python with if
Python
0 vote204 views1 answer
Iterate over rows in numpy matrix python
Python
0 vote196 views1 answer
How to find if the numpy array contains negative values
Python
0 vote193 views1 answer
Decimal characters python
Python
-1 vote188 views2 answers
Grab the first letter of each string in an array python
Python