Tensorflow图形编辑器重新路由复杂n

Question

我尝试用定制的操作来包装操作。
我解决了目标操作的输入（图片中的A），但未能包装输出。在

Init网络操作看起来很像。在

 C   D
/ \ /
B  A

假设每个运算有1个输出张量。我要加上“E”操作

^{pr2}$

我试过使用graph editor(API r0.12)，
我的策略是连接A-E和断开（C，D）-A和连接（C，D）-E

1. 通用电气连接（[E]，[C，D]）-由于op C有2个输入而失败
2. 使用ge.swap_输入2次-失败，我可以连接E和D，但仍然存在断开C-A和连接C-E的问题

我该怎么改变？
谢谢

Answer 1

您需要做一些子图remapping，以确保两个子图的签名匹配。为此，可以打印子图。在

tf.reset_default_graph()

a = tf.placeholder(dtype=tf.float32, name="a")
b = tf.placeholder(dtype=tf.float32, name="b")
c = tf.add(a, b, name="c")
d = tf.identity(a, name="d")

e = tf.identity(a, name="e")
print(ge.sgv(e.op))
print(ge.sgv(c.op, d.op).remap_inputs([0]))
ge.connect(ge.sgv(e.op), ge.sgv(c.op, d.op).remap_inputs([0]))

输出：

^{pr2}$