tf.assign的用法

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

例如：

import tensorflow as tf;
 
A = tf.Variable(tf.constant(0.0), dtype=tf.float32)
with tf.Session() as sess:
    sess.run(tf.initialize_all_variables())
    print sess.run(A)
    sess.run(tf.assign(A, 10))
    print sess.run(A)

輸出：

0.0
10.0

開始給A賦值為0，經過tf.assign函數后，把A的值變為10

==========

再例如： 

 

import tensorflow as tf

input1 = tf.placeholder(tf.float32)
input2 = tf.placeholder(tf.float32)

ouput = tf.multiply(input1,input2)

with tf.Session() as sess:
    print(sess.run(ouput,feed_dict={input1:[7.],input2:[2.]}))

輸出：

[14.]