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‘))
原文地址:https://www.cnblogs.com/reaptomorrow-flydream/p/9492550.html
时间: 2024-11-05 22:03:33