2つのグラフの構造を比較する

Question

TensorFlowには、2つのグラフが同じ構造を持っているかどうかを調べる方法はありますか？

私は、個々のインスタンスが異なるアーキテクチャを表現することが予想される抽象クラスを設計しています。私はグラフを定義するabc.abstractmethod get()を提供しました。しかし、私はまた、ディスクから事前に訓練されたグラフをロードできるようにしたい。私は事前に訓練されたグラフが具体的なクラスのget()メソッドで述べられたものと同じ定義を持っているかどうかをチェックしたいと思います。

この構造比較をどのように達成できますか？

Answer 1

現在のグラフのグラフ定義をstr(tf.get_default_graph().as_graph_def())として取得し、以前の結果と完全に等しいかどうかを比較することができます。また

、TensorFlowテストはGraphDefのプロト・オブジェクトとして、あなたは

from tensorflow.python import pywrap_tensorflow 
diff = pywrap_tensorflow.EqualGraphDefWrapper(actual.SerializeToString(), 
               expected.SerializeToString()) 
assert not diff 

を行うことができ actualと expected場合、二つのグラフは、すなわち、グラフ形式が変更された場合でも同じであることを伝えることができ、より高度な機能EqualGraphDefを持っています