小编典典

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

python

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

但是,我发现很难找到张量名称的正确语法,即使知道它的名称也很困难(而且我对操作和张量感到困惑),因此我使用张量板绘制图形并寻找名称。

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


阅读 181

收藏
2020-12-20

共1个答案

小编典典

你可以做

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

另外,如果您要在IPython笔记本中进行原型制作,则可以直接在笔记本中显示图形,请参见show_graphAlexander’s Deep
Dream笔记本中的功能

2020-12-20