天天看點

tf.add_to_collection

*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'))

結果:
[2.2, 3.0] 
5.2
注意: 
使用tf.add_n對清單元素進行相加時,清單内元素類型必須一緻,否則會報錯。
           

參考:

https://www.jianshu.com/p/6612f368e8f4

繼續閱讀