TensorFlowのテンソルインデックスの一致

Question

テンソルオブジェクトが$A_{i_0 i_1 ... i_k ... i_N}$と$B_{j_0 j_1 ... j_p ... j_M}$であると仮定します（TensorflowではそれぞれN次元とM次元の形状になります）。私は基本的に、私は

$$A_{i_0 i_1 ... 0 ... i_N} B_{j_0 j_1 ... 0 ... j_M} + A_{i_0 i_1 ... 1 ... i_N} B_{j_0 j_1 ... 1 ... j_M} + A_{i_0 i_1 ... 2 ... i_N} B_{j_0 j_1 ... 2 ... j_M}....$$ 

はどのようなこの場合の権利オプスだろう作成したい、寸法KとP以上の収縮を作成したいですか？

Answer 1

Tensorflow 11からは、einsumを使用してこれを行うことができます。

だから、あなたはあなたが行うことができますpでjを契約したいA_ijklとB_mnpを仮定し、仮定：

import tensorflow as tf 
tf.einsum("ijkl,mnj->iklmn", A,B)