如何获取Tensorflow中的所有集合？

Question

我想要获取Tensorflow模型中所有现有集合的列表。关于由Tensorflow自动创建的集合，我可以迭代GraphKeys中的所有密钥，并可以查询每个密钥，例如

tf.get_collection(tf.GraphKeys.TRAIN_OP) 

有没有办法让所有集合的完整列表，包括用户创建的集合。

Answer 1

Tensorflow将集合作为私人字典_collections存储在类Graph中。这个类还公开了一个函数来检索所有收集键/名称：

tf.get_default_graph().get_all_collection_keys()