怎麼辦行方向元素乘法tensorflow

Question

給定矩陣[[x1, x2, x3], [y1, y2, y2], [z1, z2, z3]]

我該怎麼辦[X1 * Y1 * Z1，X2 * Y2 * Z2，X3 * Y3 * Z3在tensorflow？

Answer 1

您需要使用tf.reduce_prod(x, 0)，因爲你沿着列乘數字：

import tensorflow as tf 
a = tf.constant([ 
    [1, 2, 3], 
    [4, 5, 6], 
    [7, 8, 9] 
]) 
b = tf.reduce_prod(a, 0) 

with tf.Session() as sess: 
    print sess.run(b) 

Answer 2

我認爲tf.reduce_prod是我正在尋找的。 https://www.tensorflow.org/api_docs/python/tf/reduce_prod