問題タブ [pytorch]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
python - Pytorch: スパース行列乗算
与えられた:
目標 1:
目標 2:
結果1:
結果 2:
私の PyTorch のバージョンは 0.1.12_2 です。
pytorch - テンソルとして与えられたインデックスから 1 つのホット ベクトルを作成する
4 x 64 がバッチ サイズで、6 がシーケンス長であるサイズのテンソルがあります。シーケンス ベクトルのすべての要素は、何らかのインデックス (0 から n) です。3次元のベクトルがインデックスの1つのホットエンコーディングになるテンソルを作成したい4 x 6 x n。つまり、指定されたインデックスに1を入れ、残りの値がゼロになることを意味する。
たとえば、次のテンソルがあります。
ここでは、すべての値は (0 から n) の間 (n = 15) にあります。そのため、テンソルを4 X 6 X 16、3 番目の次元が 1 つのホット エンコーディング ベクトルを表すテンソルに変換したいと考えています。
PyTorch 機能を使用してそれを行うにはどうすればよいですか? 現在、ループでこれを行っていますが、ループを避けたいです!