在Tensorflow中命名表达式的输出

Question

我想知道是否可以为某个表达式的输出命名，以便从代码的另一部分的图形中检索它。

例如

def A(a, b):
    c = a + b
    d = d * a
return d

出于调试目的，如果我可以在另一个位置拉出c而不返回整个函数层次结构，那将是很好的。

有什么想法吗？

提前致谢！

Answer 1

我假设a和b是张量。

使用c

为tf.identity命名

def A(a, b):
    c = a + b
    c = tf.identity(c, name = "c")
    d = d * a
return d

您使用tf.add操作代替+：

def A(a, b):
    tf.add(a, b, name = "c")
    d = d * a
return d

无论哪种方式，您都会使用c检索tf.get_variable('c:0')（如果有的话，您可能需要确定范围。）