03 使用Tensorflow做計算題


    我們使用Tensorflow,計算((a+b)*c)^2/a,然后求平方根。看代碼:

 1 import tensorflow as tf
 2 
 3 # 輸入儲存容器
 4 a = tf.placeholder(tf.float16)
 5 b = tf.placeholder(tf.float16)
 6 c = tf.placeholder(tf.float16)
 7 
 8 # 計算
 9 d = tf.add(a, b) #加法
10 e = tf.multiply(d, c) #乘法
11 f = tf.pow(e, 2) #平方
12 g = tf.divide(f, a) #除法
13 h = tf.sqrt(g) #平方根
14 
15 # 會話
16 sess = tf.Session()
17 
18 # 賦值
19 feed_dict= {a:1, b:2, c:3}
20 
21 # 計算
22 result = sess.run(h, feed_dict= feed_dict)
23 
24 # 關閉會話
25 sess.close()
26 
27 # 輸出結果
28 print(result)

    這里讓a=1,b=2,c=3,如果輸出9.0,證明運行成功。

    Tensorflow做計算的方法是,先把計算的式子構建一個圖,然后把這個圖和賦值在cpu上一起運行,計算速度比較快。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM