1. ホーム
  2. python

[解決済み] Tensorflowで、グラフに含まれるすべてのTensorの名前を取得する

2022-06-10 09:21:51

質問

ニューラルネットを Tensorflowskflow ;何らかの理由で、与えられた入力に対していくつかの内部テンソルの値を取得したいので、私は myClassifier.get_layer_value(input, "tensorName") , myClassifier である skflow.estimators.TensorFlowEstimator .

しかし、テンソル名を知っていても正しい構文を探すのが難しいので(演算とテンソルを混同してしまう)、tensorboardでグラフをプロットして名前を探しています。

tensorboardを使わずにグラフの中の全てのテンソルを列挙する方法はないでしょうか?

どのように解決するのですか?

以下の方法があります。

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

また、IPythonのノートブックでプロトタイピングをしている場合、ノートブックに直接グラフを表示することができます。 show_graph 関数があります。 ノートブック