功能
传入由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]]