码迷,mamicode.com
首页 > 其他好文 > 详细

TensorFlow函数(九)tf.add_to_collection()、tf.get_collection() 和 tf.add_n()

时间:2018-08-17 12:11:20      阅读:2126      评论:0      收藏:0      [点我收藏+]

标签:类型   获取   flow   loss   nbsp   collect   NPU   一个   weight   

tf.add_to_collection(name, value)

此函数将元素添加到列表中

参数:

  • name:列表名。如果不存在,创建一个新的列表
  • value:元素

tf.get_collection(name)

此函数获取列表

参数:

  • name:列表名

tf.add_n(inputs)

此函数将元素相加并返回

注意:元素类型必须一致,否者报错

1 tf.add_to_collection(losses, regularizer(weights))
2 tf.add_n(tf.get_collection(losses))

 

TensorFlow函数(九)tf.add_to_collection()、tf.get_collection() 和 tf.add_n()

标签:类型   获取   flow   loss   nbsp   collect   NPU   一个   weight   

原文地址:https://www.cnblogs.com/reaptomorrow-flydream/p/9492550.html

(1)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!