python - tensorflow-1.14,15中的操作组成

Question

我正在尝试在 tensorflow 1.15 中进行非常简单的操作。我有一个要用于推理的预训练网络。

该网络公开了一个tf.global_variablelet's sayinput和一个输出 say output。现在有一个会话变量sess，如果你想进行推理，你只需运行：

inference_output = sess.run(output, {input: input_numpy})

我的问题是我已经实现了这个推理的前端（计算input），这是一组我将调用的张量流操作，exp它需要一个输入变量 say args。

我找不到方法，因此将 的值提供exp给全局变量input，因此我的图表变为：

args -> (expr -> input) -> output

现在我这样做是：

preprocessing_inp = sess.run(expr, {args: args_numpy})
inference_output = sess.run(output, {input: preprocessing_inp})

但我希望只有一个操作（所以一切都通过一个 gpu 传递）：

inference_output = sess.run(output, {args: args_numpy})

标签： pythonpython-3.xtensorflow