indexing - Torch では、整数ラベルのリストから 1-hot tensor を作成するにはどうすればよいですか?

Question

MNIST データ セットなどから、整数クラス ラベルのバイト テンソルがあります。

 1
 7
 5
[torch.ByteTensor of size 3]

1-hot ベクトルのテンソルを作成するためにどのように使用しますか?

 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]

ループでこれを実行できることはわかっていますが、1 行で取得できる巧妙な Torch インデックスがあるかどうか疑問に思っています。

Answer 1

indices = torch.LongTensor{1,7,5}:view(-1,1)
one_hot = torch.zeros(3, 10)
one_hot:scatter(2, indices, 1)

torch/torch7 github readme (master ブランチ) でscatterドキュメントを見つけることができます。