tf.add_n([p1, p2, p3....])函数是实现一个列表的元素的相加。就是输入的对象是一个列表,列表里的元素可以是向量,矩阵,等
例如:
import tensorflow as tf;
import numpy as np;
input1 = tf.constant([1.0, 2.0, 3.0])
input2 = tf.Variable(tf.random_uniform([3]))
output = tf.add_n([input1, input2])
with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
print sess.run(input1 + input2)
print sess.run(output)输出:
[ 1.68921876 2.73008633 3.04061747]
[ 1.68921876 2.73008633 3.04061747]
本文介绍了TensorFlow中tf.add_n函数的功能与用法,该函数用于将列表中的多个张量相加。通过实例演示了如何使用tf.add_n进行操作,并展示了如何在会话中运行并获取结果。
1万+

被折叠的 条评论
为什么被折叠?



