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')
tensorflow 如何获取graph中的所有tensor name
猜你喜欢
转载自www.cnblogs.com/CheeseZH/p/12620404.html
今日推荐
周排行