Breaking News: Grepper is joining You.com. Read the official announcement!

index of max in tensor

Adorable Anteater answered on January 9, 2021 Popularity 7/10 Helpfulness 6/10

answer index of max in tensor

index of max in tensor

Comment

>>> 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)

Popularity 7/10 Helpfulness 6/10 Language python

Source: pytorch.org

Tags: max python tensor

Link to this answer
Share Copy Link

Contributed on Jan 09 2021

Adorable Anteater

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

index of max in tensor

Contents

More Related Answers

index of max in tensor

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.