9
我有一个整数类标签的字节张量,例如来自MNIST数据集。在火炬中如何从整数标签列表创建单热张量?
1
7
5
[torch.ByteTensor of size 3]
如何使用它来创建1热矢量的张量?
1 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 1 0 0 0
0 0 0 0 1 0 0 0 0 0
[torch.DoubleTensor of size 3x10]
我知道我可以用一个循环做到这一点,但我不知道是否有任何聪明的火炬索引,将让这对我在一个单一的线。