当前位置:  开发笔记 > 编程语言 > 正文

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

如何解决《在Tensorflow中,如何重命名某个操作名称?》经验,为你挑选了1个好方法。

我认为这个问题可能听起来很奇怪.比方说,有一个张量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:0或等的架构fc2/fc2wb:0.如何用特定名称包装最终操作?



1> Priyatham..:

这个答案表明tf.Graph只是附加,你不能修改它.

话虽这么说,重新命名张量的一个更好的方法是使用这样的tf.identity:

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

编辑:在找到我自己的一个问题的答案后,我可以看到一个更好的方法来做到这一点:

graph_def.node[].name = "out"

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

推荐阅读
手机用户2402852307
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有