[pytorch] RuntimeError: one_hot is only applicable to index tensor

 

 

import torch.nn.functional as F
bs = 10
F.one_hot(torch.ones(bs)*2, 5)

2에 대응되는 전체 크기는 5인 one hot encoding vector를 batch size개수만큼 만들고 싶어서 위와 같이 코드를 작성하면 런타임 에러가 뜬다.

 

RuntimeError: one_hot is only applicable to index tensor.

one_hot 함수는 index tensor (타입 명이 torch.int64)를 입력으로 받기 때문에 발생하는 에러이다. 다음과 같이 수정해주면 된다.

 

>>> torch.nn.functional.one_hot((torch.ones(bs)*2).to(torch.int64), 5)
tensor([[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0]])

 

 

 

끝 ◼︎