如何在张量流中进行逐行元素乘法

Question

给定矩阵renrem

如何在张量流中做[x1 * y1 * z1，x2 * y2 * z2，x3 * y3 * z3]？

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