我正在尝试获取Tensorflow模型中所有现有集合的列表.关于Tensorflow自动创建的集合,我可以迭代所有键,GraphKeys并可以查询每个键,例如
tf.get_collection(tf.GraphKeys.TRAIN_OP)
有没有办法获得所有集合的完整列表,包括用户创建的集合.
blu*_*e10 10
Tensorflow将集合存储为_collections类中的私有字典Graph.该类还公开了一个函数来检索所有集合键/名称:
tf.get_default_graph().get_all_collection_keys()