tf.add_to_collection、tf.get_collection、tf.add_n用法の浅い分析
2484 ワード
例:
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リスト要素を加算する場合は、リスト内の要素タイプが一致する必要があります.そうしないと、エラーが発生します.