問題タブ [numpy-einsum]
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.
207 問題
0
投票する
1
に答える
271
参照
numpy - 順次積と和を 3D でのより高速な行列演算に置き換える
私の現在の theano スクリプトでは、ボトルネックは次のコードです。
2D の場合、elementwise+sum を内積に置き換えると 5 倍のスピードアップが見られました。この場合、私を助けることができる行列演算はありますか?
編集：
Divakarは私にeinsumに基づくバージョンをくれました。ただし、私の意図はそれをtheanoに移植することであり、einsumはtheanoではサポートされていません。したがって、theanoに移植可能な代替手段は大歓迎です。