2017-04-18 12 views
2

Tensorflowモデル内の既存のすべてのコレクションのリストを取得しようとしています。 Tensorflowによって自動的に作成されたコレクションについては、GraphKeysのすべてのキーを繰り返し処理でき、各キーを照会することができます。 Tensorflowですべてのコレクションを取得するには?

tf.get_collection(tf.GraphKeys.TRAIN_OP) 

は、ユーザが作成したコレクションを含むすべてのコレクションの完全なリストを取得する方法があります。

答えて

7

Tensorflowは、コレクションをプライベート辞書_collectionsとしてクラスGraphに格納します。このクラスは、すべてのコレクションキー/名前を取得する関数も公開しています。

tf.get_default_graph().get_all_collection_keys() 
関連する問題