1. 程式人生 > >tf.add_n函式的用法

tf.add_n函式的用法

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]