Numpyのeinsumを使って３次元配列の掛け算 - Qiita

知り合いに簡単に3次元配列の掛け算をする方法を聞かれ、調べてみたらNumpyのライブラリにアインシュタインの縮約記号を用いたeinsumというのがありこれを使えば楽にできそうなのでやってみました。 １ｘ２ｘ５の3次元行列と２ｘ３ｘ５の三次元行列をまず適当に2つ作ります。 A = \left( \begin{matrix} a_0 & a_1 \end{matrix} \right) , a_0 = \left[ \begin{matrix} 0 & 1 & 1 & 0 & 0 \end{matrix} \right] , a_1 = \left[ \begin{matrix} 0 & 1 & 0 & 0 & 1 \end{matrix} \right] W = \left( \begin{matrix} w_0 & w_1 & w_2 \\ w_3 & w_4 & w_5 \end{mat