如何从有限列表中传递元素作为输入？

Question

我有“石头剪刀布”游戏的状态列表。 我计划对这些类别进行编码，如下所示：

[1, 0, 0] - rock 
[0, 1, 0] - paper 
[0, 0, 1] - Rscissors

有方便的自动方式吗？

我使用了Embedding层，但我不确定它是否合适。

Answer 1

index = 2
categories_length = 3

tensor = torch.nn.functional.one_hot(
    torch.LongTensor([index]),
    categories_length,
)