我正在创建神经网络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(使用前将#替换为@)