獲取所有variable(每個op中可訓練的張量)的name:
1
2
|
for variable_name in tf.global_variables(): print (variable_name) |
獲取所有tensor(每個op的輸出張量)的name:
1
2
|
for tensor_name in tf.contrib.graph_editor.get_tensors(tf.get_default_graph()): print (tensor_name) |
獲取所有op及其輸入輸出的name:
1
2
3
|
with tf.Session() as sess: for node in sess.graph_def.node: print (node) |
以上這篇tensorflow 獲取所有variable或tensor的name示例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/yuejisuo1948/article/details/88858190