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

1 vote253 views1 answer
Count missing values groupby
Python
1 vote265 views1 answer
Python create a matrix with one in diagonal
Python
1 vote249 views1 answer
Python psycopg2 utf8
Python
0 vote267 views1 answer
Python make an object hashable
Python
2 votes250 views1 answer
Python read excel set index
Python
2 votes224 views1 answer
Check if camera is being used python
Python
0 vote258 views6 answers
Django wht post save signal not firing
Python
0 vote250 views1 answer
Ipython widget display
Python
3 votes269 views1 answer
Create or update django models
Python
0 vote259 views1 answer
Pandas add quantile columns
Python