問題タブ [numpy-ufunc]

質問する

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.

172 問題
Newest
Active
Bountied
318
Unanswered
0 投票する
1 に答える
507 参照

gpu - Numba で結合されたベクトル化関数

GPU のコードを最適化するために Numba (バージョン 0.37.0) を使用しています。ベクトル化された関数を組み合わせて使用​​したい (Numba の @vectorize デコレータを使用)。

インポートとデータ:

これは期待どおりに機能します。

出力：

しかし、別のベクトル化された関数を呼び出して、「l2_norm」内でその「for」を使用することは避けたいと思います。

私はこれを試しました：

ただし、TypingError が発生します。

この組み合わせを実行する方法について何か考えはありますか?

Numba で l2_norm を最適化する他の方法について何か提案はありますか?


12345678910