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

tf.add_to_collection(name, value)

此函數將元素添加到列表中函數

參數:

  • name:列表名。若是不存在,建立一個新的列表
  • value:元素

tf.get_collection(name)

此函數獲取列表spa

參數:

  • name:列表名

tf.add_n(inputs)

此函數將元素相加並返回code

注意:元素類型必須一致,否者報錯

1 tf.add_to_collection('losses', regularizer(weights))
2 tf.add_n(tf.get_collection('losses'))
相關文章
相關標籤/搜索