在Tensorflow中,获取图中所有Tensors的名称

2023-11-27

我正在创建神经网络Tensorflow and skflow;由于某种原因,我想获取给定输入的一些内部张量的值,所以我使用myClassifier.get_layer_value(input, "tensorName"), myClassifier成为一个skflow.estimators.TensorFlowEstimator.

然而,我发现很难找到张量名称的正确语法,即使知道它的名称(并且我在操作和张量之间感到困惑),所以我使用张量板来绘制图形并查找名称。

有没有一种方法可以在不使用张量板的情况下枚举图中的所有张量?


你可以做

[n.name for n in tf.get_default_graph().as_graph_def().node]

另外,如果您在 IPython 笔记本中进行原型设计,则可以直接在笔记本中显示图形,请参阅show_graph亚历山大深梦中的功能notebook

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在Tensorflow中,获取图中所有Tensors的名称 的相关文章

随机推荐