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

Question

我正在尝试获取Tensorflow模型中所有现有集合的列表。关于Tensorflow自动创建的集合，我可以遍历GraphKeys中的所有键，并可以查询每个键，例如：

 tf.get_collection(tf.GraphKeys.TRAIN_OP)

有没有一种方法可以获取所有收藏夹的完整列表，包括用户创建的收藏夹？

- bluenote10

Answer 1

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

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

tf.get_default_graph().get_all_collection_keys()

- bluenote10