例子源码地址:
https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/1%20-%20Introduction/basic_operations.py
根据网上的入门例子,一点点的熟悉代码和TensorFlow。对这个基本的例子,做一个注释,备忘之余分享给同样入门的初学者。
import tensorflow as tf a = tf.constant(2) b = tf.constant(3)
#把a,b定义为tensorflow的常量,并且赋值。
with tf.Session() as sess: print "a=2, b=3" print "Addition with constants: %i" % sess.run(a+b) print "Multiplication with constants: %i" % sess.run(a*b)
#使用Session的run()输出a+b和a*b的结果。使用with tf.Session() as sess这种用法,在sess.run()结束之后,不用调用sess.close()释放资源。
a = tf.placeholder(tf.int16) b = tf.placeholder(tf.int16)
#把a,b定义成为tf.int16类型的占位符,并没有放具体的数值进去。
add = tf.add(a, b) mul = tf.mul(a, b)
#在a,b两个占位符之上,定义了两个操作,add和mul。
with tf.Session() as sess: print "Addition with variables: %i" % sess.run(add, feed_dict={a: 2, b: 3}) print "Multiplication with variables: %i" % sess.run(mul, feed_dict={a: 2, b: 3})
#给a,b赋值,并且对他们进行前文定义的add和mul操作。
matrix1 = tf.constant([[3., 3.]]) matrix2 = tf.constant([[2.],[2.]]) product = tf.matmul(matrix1, matrix2)
#定义两个常量矩阵,并且为这两个矩阵定义了一个matmul操作product。
with tf.Session() as sess: result = sess.run(product) print result
#运行product操作,并且将结果result输出。