one_hot is only applicable to index tensor.

Clear Copperhead answered on August 11, 2022 Popularity 8/10 Helpfulness 7/10

answer one_hot is only applicable to index tensor.

Comment

torch.nn.functional.one_hot(my_tensor.long())  # <- make sure its int64

Popularity 8/10 Helpfulness 7/10 Language python

Source: github.com

Tags: python

Link to this answer
Share Copy Link

Contributed on Aug 11 2022

Clear Copperhead

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.

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