求和积和常数在张量流中

Question

我正在尝试在tensorflow中进行以下计算

Y = X1*W1 + X2*W2 + X3*W3 + b

X和W具有相同的形状，X * W是点积。 玩了一会儿后，我发现下面这件作品

Y = tf.add(tf.add(tf.add(tf.reduce_sum(tf.multiply(X1,W1)),tf.reduce_sum(tf.multiply(X2,W2))),tf.reduce_sum(tf.multiply(X3,W3))),b)

在tensorflow中执行此操作的简洁方法是什么？

Answer 1

如何定义dotproduct函数？

def dotprod(X,W):
  return tf.reduce_sum(tf.multiply(X,W))
Y = dotprod(X1,W1) + dotprod(X2,W2) + dotprod(X3,W3) + b

我相信您可以使用+来添加张量。

让我知道这是否有效！