*tf.add_to_collection(‘list_name’, element):將元素element添加到列表list_name中
*tf.get_collection(‘list_name’):返回名稱為list_name的列表
*tf.add_n(list):將列表元素相加并返回
import tensorflow as tf
tf.add_to_collection('losses', tf.constant(2.2))
tf.add_to_collection('losses', tf.constant(3.))
with tf.Session() as sess:
print(sess.run(tf.get_collection('losses')))
print(sess.run(tf.add_n(tf.get_collection('losses'))
結(jié)果:
[2.2, 3.0]
5.2
注意:
使用tf.add_n對列表元素進行相加時谒兄,列表內(nèi)元素類型必須一致,否則會報錯。