1. 程式人生 > >tf.assign的用法

tf.assign的用法

tf.assign的用法

原創  2017年05月16日 08:26:16
  • 13943

tf.assign(A, new_number): 這個函式的功能主要是把A的值變為new_number

例如:

[python]  view plain  copy
  1. import tensorflow as tf;  
  2.   
  3. A = tf.Variable(tf.constant(0.0), dtype=tf.float32)  
  4. with tf.Session() as sess:  
  5.     sess.run(tf.initialize_all_variables())  
  6.     print sess.run(A)  
  7.     sess.run(tf.assign(A, 10
    ))  
  8.     print sess.run(A)  
輸出:

0.0
10.0

開始給A賦值為0,經過tf.assign函式後,把A的值變為10