tensorflow 如何獲取graph中的所有tensor name


import tensorflow as tf

saved_model_dir = "./saved_model"

with tf.Session(graph=tf.Graph()) as sess:
    tf.saved_model.loader.load(sess, ["serve"], saved_model_dir)
    graph = tf.get_default_graph()
    [print(n.name) for n in tf.get_default_graph().as_graph_def().node]

# 得到name之后,就可以獲取相應的tensor了,例如:
# input_tensor = sess.graph.get_tensor_by_name('input:0')
# output_tensor = sess.graph.get_tensor_by_name('output:0')


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM