tf.math.add_n()

功能

传入由tensor组成的list,将所有的tensor做加法操作,要求所有的tensor 形状相同(不支持numpy广播)

别名

  • tf.add_n 最常用

语法

tf.math.add_n(
    inputs,
    name=None
)
参数
  • inputs:有tensor组成的list
  • name:
返回值

tensor,形状与输入的tensor相同

例子

a=tf.constant(np.arange(6).reshape(2,3))
b=tf.constant(np.arange(3,9).reshape(2,3))

print("a:",a.numpy(),sep="\n")
print("b:",b.numpy(),sep="\n")

res=tf.math.add_n([a,b])
print("res:",res.numpy(),sep="\n")

执行结果:

a:
[[0 1 2]
 [3 4 5]]
b:
[[3 4 5]
 [6 7 8]]


res:
[[ 3  5  7]
 [ 9 11 13]]

原文出处:/show_1EF4jL2wPDjG.html