TF Variable1

1. tensorflow demo

 1 import tensorflow as tf
 2  
 3  
 4 # 定义一个变量
 5   
 6 t = tf.Variable(1,name='t')
 7   
 8 # 初始化变量
 9 
10 sess = tf.Session()
11  
12 sess.run(tf.global_variables_initializer())
13  
14 # 打印当前值
15  
16 sess.run(t)
17  
18 # 保存当前值
19  
20 saver = tf.train.Saver([t])
21  
22 saver.save(sess,'/tmp/t.chpt',global_step=0)
23  
24  
25 # 修改值
26  
27 sess.run(tf.assign_add(t,2))
28  
29  
30 # 恢复
31 saver.restore(sess,'/tmp/t.chpt-0')

  

猜你喜欢

转载自www.cnblogs.com/nika86/p/10322022.html
今日推荐