問題タブ [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.

0 投票する
1 に答える
6171 参照

python - Pytorch: スパース行列乗算

与えられた:

目標 1:

目標 2:

結果1:

結果 2:

私の PyTorch のバージョンは 0.1.12_2 です。

0 投票する
3 に答える
15593 参照

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 機能を使用してそれを行うにはどうすればよいですか? 現在、ループでこれを行っていますが、ループを避けたいです!