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 vote100 views1 answer
Python threading vs multiprocessing
Python
0 vote99 views1 answer
Line plotly with shaded area
Python
14 votes114 views1 answer
Sort list of dictionaries by key python
Python
14 votes93 views3 answers
Python remove first and last character from string
Python
0 vote106 views1 answer
For an HTML output, you don’t need an additional library. It simply goes like this:
Python
0 vote106 views4 answers
Python call function in class
Python
0 vote103 views1 answer
Strip unicode characters from strings python
Python
0 vote104 views1 answer
Python get ids from array of objects
Python
0 vote101 views1 answer
Which game engine is best
Python
0 vote107 views1 answer
Converting 4hr 20min to minutes
Python