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