ナンシー：エインサムのないテンソルアダマール乗算はできますか？

Question

私はランク4のテンソルA（例えば、インデックス（a、b、i、j））とランク2のテンソルB（インデックス（i、j））を得ています。それら。

つまり、商品Cを呼び出すと、C[a,b,i,j] == A[a,b,i,j] * B[i,j]が必要です。 einsumでこれを行うにはかなり単純な方法がありますが、tensordotなどのメソッドと比較して、einsumを使用するとかなりのパフォーマンスが低下すると言われていますが、ドキュメントから避けるための良い方法が見つかりませんでした。

私はそれを逃した可能性があります。私はテンソルに慣れていてnumpyの専門家ではありません。

Answer 1

C = A * B 

broadcasting rules後、numpyのは、最後の軸から始めAとBの形状をラインアップします。

A: (a, b, i, j) 
B:  (i, j) 

をしてC作成するために一緒にAとBの対応する要素を乗算：

を

A[a, b, i, j] 
* B[  i, j] 
== C[a, b, i, j]