テンソルフローは行列の行列を使用できますか？

Question

要素がプリミティブデータ型（整数または浮動小数点数）の場合、テンソルフローは[ [a,b,c] ] x [ [x],[y],[z] ]のような式を計算できます。 a,bおよびcが1x3行列で、x,yおよびzが3x1行列の場合、同様の計算を実行できますか？

TensorFlowはこの式を計算して最適化できますか？

Answer 1

tf.batch_matmul()演算子は、行列のバッチに対して行列乗算を実行できます。この場合、テクスチャabcの形は(3, 1, 3)（abc[0, :, :] = a,abc[1, :, :] = bなど）、テンソルxyzの形は(3, 3, 1)（xyz[0, :, :] = xなど）です。

abc = ... 
xyz = ... 

result = tf.batch_matmul(abc, xyz) 

print result.get_shape() # ==> "(3, 1, 1)" 

resultはtf.pack([tf.matmul(a, x), tf.matmul(b, y), tf.matmul(c, z)])と同等の内容を3次元テンソルです。