tf.add_to_collection、tf.get_collection、tf.add_n用法浅析

来源:互联网 发布:机顶盒检测软件 编辑:程序博客网 时间:2024/05/21 06:34
  • tf.add_to_collection(‘list_name’, element):将元素element添加到列表list_name中
  • tf.get_collection(‘list_name’):返回名称为list_name的列表
  • tf.add_n(list):将列表元素相加并返回

例子:

import tensorflow as tftf.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对列表元素进行相加时,列表内元素类型必须一致,否则会报错。

原创粉丝点击