Code Error

[awei-lwj]

In the chapter02

import torch import torch.nn.functional as F 
input_ids = torch.tensor(input_ids) 
one_hot_encodings = F.one_hot(input_ids, num_classes=len(token2idx)) 



one_hot_encodings.shape 
torch.Size([38, 20])

[KevinZonda]

be54277b278d5044ad6e3d9e3ef0fe323f838fd1

[znsoftm]

Thanks. Has been merged.

[KevinZonda]

@znsoftm Hi, should close this issue