比较两张图的结构

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