您好,登錄后才能下訂單哦!
TensorFlow修改變量值后,需要重新賦值,assign用起來有點小技巧,就是需要需要弄個操作子,運行一下。
下面這么用是不行的
import tensorflow as tf import numpy as np x = tf.Variable(0) init = tf.initialize_all_variables() sess = tf.InteractiveSession() sess.run(init) print(x.eval()) x.assign(1) print(x.eval())
正確用法
1.
import tensorflow as tf x = tf.Variable(0) y = tf.assign(x, 1) with tf.Session() as sess: sess.run(tf.global_variables_initializer()) print sess.run(x) print sess.run(y) print sess.run(x)
2.
In [212]: w = tf.Variable(12) In [213]: w_new = w.assign(34) In [214]: with tf.Session() as sess: ...: sess.run(w_new) ...: print(w_new.eval()) # output 34
3.
import tensorflow as tf x = tf.Variable(0) sess = tf.Session() sess.run(tf.global_variables_initializer()) print(sess.run(x)) # Prints 0. x.load(1, sess) print(sess.run(x)) # Prints 1.
我的方法
import numpy as np #這是Python的一種開源的數值計算擴展,非常強大 import tensorflow as tf #導入tensorflow ##構造數據## x_data=np.random.rand(100).astype(np.float32) #隨機生成100個類型為float32的值 y_data=x_data*0.1+0.3 #定義方程式y=x_data*A+B ##-------## ##建立TensorFlow神經計算結構## weight=tf.Variable(tf.random_uniform([1],-1.0,1.0)) biases=tf.Variable(tf.zeros([1])) y=weight*x_data+biases w1=weight*2 loss=tf.reduce_mean(tf.square(y-y_data)) #判斷與正確值的差距 optimizer=tf.train.GradientDescentOptimizer(0.5) #根據差距進行反向傳播修正參數 train=optimizer.minimize(loss) #建立訓練器 init=tf.global_variables_initializer() #初始化TensorFlow訓練結構 #sess=tf.Session() #建立TensorFlow訓練會話 sess = tf.InteractiveSession() sess.run(init) #將訓練結構裝載到會話中 print('weight',weight.eval()) for step in range(400): #循環訓練400次 sess.run(train) #使用訓練器根據訓練結構進行訓練 if step%20==0: #每20次打印一次訓練結果 print(step,sess.run(weight),sess.run(biases)) #訓練次數,A值,B值 print(sess.run(loss)) print('weight new',weight.eval()) #wop=weight.assign([3]) #wop.eval() weight.load([1],sess) print('w1',w1.eval())
以上這篇對TensorFlow的assign賦值用法詳解就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。