2つの3Dマトリクス上の2次元のドットマトリクスに沿った

Question

それぞれ（x、y、z）および（y、x、z）の寸法を有する2つのマトリクスAおよびBが与えられ、 2つの行列？結果の次元は（x、x、z）でなければなりません。

ありがとうございます！

Answer 1

文字通り同じ文字列式で使用np.einsum - そのiがで新たな次元であると考えられるとして、我々は、二番目の配列ではなくyxzのための文字列表記としてyizを使用していた

np.einsum('xyz,yiz->xiz',a,b) # a,b are input arrays 

注意xが割り当てられている最初の配列の最初の軸に揃えてはなりません。整列されるディメンションには同じ文字列表記が与えられます。