tf.add_to_collection、tf.get_collection、tf.add_n用法の浅い分析

2484 ワード

  • tf.add_to_collection(‘list_name’,element):リストリストに要素elementを追加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リスト要素を加算する場合は、リスト内の要素タイプが一致する必要があります.そうしないと、エラーが発生します.