如何在TensorFlow图的开头"追加"Op？

Question

我有一个GraphDef我正在导入的原型文件tf.import_graph_def.可以在图的末尾添加Ops,如下所示:

final_tensor = tf.import_graph_def(graph_def, name='', return_elements=['final_tensor'])
new_tensor = some_op(final_tensor)

但是我想在图的开头添加Ops,所以graph_def中的第一个Op基本上需要将我的Op的输出作为输入,我该怎么做？

Answer 1

终于找到了办法做到这一点.我确信评论中提到的Yarolsav函数在内部做了类似的事情.

new_input = graph_def.node.add()
new_input.op = 'new_op_name'  # eg: 'Const', 'Placeholder', 'Add' etc
new_input.name = 'some_new_name'
# set any attributes you want for new_input here
old_input.input[0] = 'some_new_name'  #  must match with the name above

有关如何设置属性的详细信息,请参阅此文件.