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 및 페이지 이상 수축을 만들려면?

Answer 1

Tensorflow 11부터는 einsum을 사용하면됩니다.

그래서 A_ijkl 및 B_mnp을 가정하고 당신이 할 수있는 p와 j 계약 할 랬 :

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