問題タブ [jax]
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.
jit - jax.numpy.unique を使用する関数を jit することは可能ですか?
次のコードは機能しません。
エラーメッセージは、次の使用について説明していますjnp.unique。
シャープ ビットに関するドキュメントでは、内部配列の形状が引数の値に依存する場合、jit は機能しないと説明されています。これはまさにここに当てはまります。
ドキュメントによると、潜在的な回避策は静的パラメーターを指定することです。しかし、これは私の場合には当てはまりません。パラメータは、ほぼすべての関数呼び出しで変更されます。コードを、このような計算を実行する前処理ステップjnp.uniqueと、jit できる計算ステップに分割しました。
それでも聞きたいのですが、私が気付いていない回避策はありますか？
python - jax 配列を使用した numpy 配列へのインデックス付け: 誤ったエラー メッセージ
次の numpy コードはまったく問題ありません。
jax に移行した後も機能します。
それでは、numpy と jax の組み合わせを試してみましょう。
これにより、次のエラーが発生します。
jax 配列を使用した numpy 配列へのインデックス付けがサポートされていない場合、それは問題ありません。しかし、エラーメッセージは間違っているようです。そして、事態はさらに混乱します。形状を少し変更すると、コードは正常に機能します。次のサンプルでは、(30,) から (40,) までのインデックスの形状のみを編集しました。エラーメッセージはもうありません:
CPUでjaxバージョン「0.2.12」を実行しています。ここで何が起きてるの？