
我们使用Tensorflow,计算((a+b)*c)^2/a,然后求平方根。看代码:
import tensorflow as tf # 输入储存容器
a = tf.placeholder(tf.float16)
b = tf.placeholder(tf.float16)
c = tf.placeholder(tf.float16) # 计算
d = tf.add(a, b) #加法
e = tf.multiply(d, c) #乘法
f = tf.pow(e, 2) #平方
g = tf.divide(f, a) #除法
h = tf.sqrt(g) #平方根 # 会话
sess = tf.Session() # 赋值
feed_dict= {a:1, b:2, c:3} # 计算
result = sess.run(h, feed_dict= feed_dict) # 关闭会话
sess.close() # 输出结果
print(result)
这里让a=1,b=2,c=3,如果输出9.0,证明运行成功。
Tensorflow做计算的方法是,先把计算的式子构建一个图,然后把这个图和赋值在cpu上一起运行,计算速度比较快。