1. 程式人生 > >tensorflow:tf.GraphKeys

tensorflow:tf.GraphKeys

  • **圖集合。**TensorFlow提供了一種通用機制,以便在tf.Graph中儲存元資料集合。tf.add_to_collection函式允許您將物件列表與一個鍵相關聯(其中tf.GraphKeys定義了部分標準鍵),tf.get_collection則允許您查詢與鍵關聯的所有物件。TensorFlow庫的許多組成部分會使用它:例如,當您建立tf.Variable時,系統會預設將其新增到表示“全域性變數(tf.global_variables)”和“可訓練變數tf.trainable_variables)”的集合中。當您後續建立tf.train.Savertf.train.Optimizer
    時,這些集合中的變數將用作預設引數。