3d times 2d行列Tensorflow

Question

異なるテンソルの2つのテンソルがあるとします。

<tf.Tensor 'embedding_lookup:0' shape=(?, 128) dtype=float32> 
<tf.Tensor 'Neg:0' shape=(?, 15, 128) dtype=float32> 

私は形(?, 15, 1)か(?, 15)と、最終的なテンソルを得るようなこれら二つのテンソルの行列の乗算をしたいです。どうすればこのことができますか？出力は(?, 15)の形状を持っているでしょう

tf.einsum('ij,ikj->ik',X,Y) 

-

Answer 1

それぞれ2Dと3D入力としてXとYで、我々はtf.einsumを使用することができます。