张量流中的剪枝计算图

Question

tensorflow中的计算图有一些节点input和一些节点output。 input表示神经网络的输入张量，output它的预测。 input的其他一些子图提要，output跟随一些损失（以及其他计算）。培训结束后，我只想在input和output之间保留（前向）子图。是否可以删除所有其他操作？ Deleting all but a few nodes in TensorFlow graph表示这是不可能的。也许通过使用tf.train.export_meta_graph

导出内容来解析

Answer 1

虽然无法从tf.Graph删除节点，但TensorFlow包含的工具可以修剪较低级别的tf.GraphDef协议缓冲区。 TensorFlow graph transformation tool包含一个名为strip_unused_nodes的转换过程，允许您命名输入节点集和一组输出节点，并返回已修剪的tf.GraphDef。