gpt4 book ai didi

python - 在Tensorflow中,如何重命名某个操作名称?

转载 作者:太空狗 更新时间:2023-10-29 20:53:24 26 4
gpt4 key购买 nike

我觉得这个问题听起来很奇怪。比方说,有一个张量 a

x = tf.placeholder(tf.float32, [None, 2400], name="x")
y = tf.placeholder(tf.float32, [None, 2400], name="y")
a = tf.add(x, y, name="a")

是否有一种有效的方法来引用具有不同名称的a,例如out?我在想像这样的虚拟操作

b = tf.add(a, 0, name="out")

我问这个是因为我正在尝试不同的网络架构,无论架构如何,我都想访问具有一致名称的输出张量,例如

tf.get_default_graph().get_tensor_by_name("out:0")`

目前,输出张量取决于 fc1/fc1wb:0fc2/fc2wb:0 等架构。如何用特定名称包装最终操作?

最佳答案

This答案表明 tf.Graph 只是附加的,你不能修改它。

话虽如此,重命名张量的一种稍微好一点的方法是使用 tf.identity像这样:

tf.identity(a, name="out")

编辑:在找出我自己的一个问题的答案后,我可以找到一种稍微好一点的方法:

graph_def.node[<index of the op you want to change>].name = "out"

index 通常是 -1 因为最终输出张量在 graph_def

的末尾

关于python - 在Tensorflow中,如何重命名某个操作名称?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41819397/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com