pytorch one_hot 实例

import torch

a = torch.FloatTensor(2, 5).zero_()
b = a.scatter_(dim=1,index=torch.LongTensor([[3],[4]]),value=1)
print(b)

a = torch.FloatTensor(2, 5).zero_()
b = a.scatter_(dim=1,index=torch.LongTensor([[2],[4]]),value=1)
print(b)

a = torch.FloatTensor(2, 5).zero_()
b = a.scatter_(dim=1,index=torch.LongTensor([[3],[4]]),value=2)
print(b)

0 0 0 1 0
0 0 0 0 1
[torch.FloatTensor of size 2x5]

0 0 1 0 0
0 0 0 0 1
[torch.FloatTensor of size 2x5]

0 0 0 2 0
0 0 0 0 2
[torch.FloatTensor of size 2x5]