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


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

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM