首页 > 技术文章 > tensorflow

cyx-garen 2018-08-08 13:42 原文

读取
sess.run(tf.get_default_graph().get_tensor_by_name('test:0')

saver.restore(sess,'/ckpt/rnn.ckpt')
graph = tf.get_default_graph()
pred_y = tf.get_collection('pred_y')[0]
X = graph.get_operation_by_name('X').outputs[0]    X为placeholder  所以使用 operation_by_name
Y = graph.get_operation_by_name('Y').outputs[0]



一定要注意 saver=tf.train.Saver() 的位置 在所有变量定义后! 大坑

  

推荐阅读