pytorchでonHot化

Question

[0,0,1,2,3,1]

を

[1,0,0,0]
[1,0,0,0]
[0,1,0,0]
[0,0,1,0]
[0,0,0,1]
[0,1,0,0]

にしたいです

pytorchを使っています

onHot化するにはどうすればいいですか？わかる人教えてください
追記
横幅が変動しないものがいいです
x番目を一にするもの　というほうが近いです
結構大きいデー多を何回もやるので　
forとかnumpyに変換してうんうんとかじゃ遅すぎるので
できるだけ軽いものがいいです

Comments

[jbpb0]

これ？
https://qiita.com/whale9490/items/ade0c201b9d51846dadf

Answer 1

jbpbさんのurlで解決しました https://qiita.com/whale9490/items/ade0c201b9d51846dadf

Answer 2

forとかnumpyに変換してうんうんとかじゃ遅すぎるので
できるだけ軽いものがいいです

とのことですが、numpyは、高速なライブラリを呼び出しているので、numpyなしでやるよりずっと高速です。

作り方は、以下の

NumPyのeyeまたはidentityでone-hot表現に変換

をご覧ください。

Comments

[jbpb0]

リンク間違ってます

[ppaul]

ありがとうございます。修正しました。

[Flan.]

pytorchでお願いします　torchからnumpyに変換して戻す工程が　という話です　一応試してはみます