2
我想要獲取Tensorflow模型中所有現有集合的列表。關於由Tensorflow自動創建的集合,我可以迭代GraphKeys
中的所有密鑰,並可以查詢每個密鑰,例如如何獲取Tensorflow中的所有集合?
tf.get_collection(tf.GraphKeys.TRAIN_OP)
有沒有辦法讓所有集合的完整列表,包括用戶創建的集合。
我想要獲取Tensorflow模型中所有現有集合的列表。關於由Tensorflow自動創建的集合,我可以迭代GraphKeys
中的所有密鑰,並可以查詢每個密鑰,例如如何獲取Tensorflow中的所有集合?
tf.get_collection(tf.GraphKeys.TRAIN_OP)
有沒有辦法讓所有集合的完整列表,包括用戶創建的集合。
Tensorflow將集合作爲私人字典_collections
存儲在類Graph
中。這個類還公開了一個函數來檢索所有收集鍵/名稱:
tf.get_default_graph().get_all_collection_keys()