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