首页 > 代码库 > 03 使用Tensorflow做计算题
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做计算的方法是,先把计算的式子构建一个图,然后把这个图和赋值传给C++写的程序一起计算,比较快。
03 使用Tensorflow做计算题
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。