Сравнение структур двух графиков

Question

Есть ли способ в TensorFlow узнать, имеют ли две диаграммы одинаковые структуры?

Я проектирую абстрактный класс, отдельные экземпляры которого должны представлять различные архитектуры. Я предоставил abc.abstractmethod get(), который определяет график. Тем не менее, я также хочу иметь возможность загружать предварительно подготовленный график с диска. Я хочу проверить, имеет ли предварительно подготовленный график то же определение, что и в методе get() конкретного класса.

Как достичь такого структурного сравнения?

Answer 1

Вы можете получить график определения текущего графика как str(tf.get_default_graph().as_graph_def()) и сравнить его точное соответствие с вашим предыдущим результатом.

Кроме того, TensorFlow тесты имеют более продвинутые функции EqualGraphDef, которые могут сказать, что два графа равны, даже если формат график изменился, то есть, если actual и expected как объекты GraphDef прото, вы могли бы сделать

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