看代码, 学TF(一): 先动手写第一段代码

时间:2022-11-22 09:28:18

我们直接来看代码, 不要怕.稍后会有逐行解释.

[python] view plain copy print ?
  1. import tensorflow as tf  
  2.   
  3. node1 = tf.constant(3.0, tf.float32)  
  4. node2 = tf.constant(4.0)  
  5. node3 = tf.add(node1, node2)  
  6.   
  7. sess = tf.Session()  
  8. print “node3:”, node3  
  9. print “sess.run(node3): ”, sess.run(node3)  
  10.   
  11. a = tf.placeholder(tf.float32)  
  12. b = tf.placeholder(tf.float32)  
  13. adder_node = a + b  
  14.   
  15. print sess.run(adder_node, {a: 3, b: 4.5})  
  16. print sess.run(adder_node, {a: [13], b: [24]})  
  17.   
  18. add_and_triple = adder_node * 3.  
  19. print sess.run(add_and_triple, {a: 3, b: 4.5})  
  20.   
  21. W = tf.Variable([.3], tf.float32)  
  22. b = tf.Variable([-.3], tf.float32)  
  23. x = tf.placeholder(tf.float32)  
  24. linear_model = W * x + b  
  25.   
  26. init = tf.global_variables_initializer()  
  27. sess.run(init)  
  28. print sess.run(linear_model, {x: [1234]})  
  29.   
  30. y = tf.placeholder(tf.float32)  
  31. squares_deltas = tf.square(linear_model - y)  
  32. loss = tf.reduce_sum(squares_deltas)  
  33. print sess.run(loss, {x:[1234], y:[0, -1, -2, -3]})  
  34.   
  35. fixW = tf.assign(W, [-1.])  
  36. fixb = tf.assign(b, [1.])  
  37. sess.run([fixW, fixb])  
  38. print sess.run(loss, {x: [1,2,3,4], y: [0,-1,-2,-3]})  
看代码, 学TF(一): 先动手写第一段代码

两个概念

在分析代码前,我们需要弄懂两个tensorflow的概念, 往下看,这并不难.

1. tensor(张量)

矩阵,我想大家知道是什么,不知道的话,可以认为他就是二维数组.而张量可以认为是矩阵的推广,简单来说,他可以表示更高维的数组.当然包括1维,2维都可以表示.
举些例子,下面都是张量:

  • [1] : 这是最简单的张量, 其实也就是标量
  • [1,2,3,4]: 这是1维数组,也是张量
  • [[1,2,3,4], [11,22,33,44]] :这是2维数组,从矩阵的角度可以认为是一个2X4的矩阵,即2行,4列

上面的张量,你可能还是比较熟悉的,我们来看下面这个,仔细观察一下:

  • [ [ [ ], [ ] , [ ] , [ ] ] , [ [ ] , [ ] , [ ] , [ ] ] ]
    没错,聪明的你可能已经发现这其实就是上面那个2维数组中的每一个数字都替换成[],也就是替换成一个空数组,现在,我们让这每一个空数组里都加3个变量,变成:
  • [[[1,1,1], [2,2,2], [1,2,3], [3,2,1]], [[2,2,2], [3,3,3], [4,4,4], [5,5,5]]]

2. node(节点)

代码分析

第一行

import tensorflow as tf

导入tensorflow库, 并且定义了一个别名”tf”.使得代码可以用tf.的方式使用tensorflow定义的一切类, 函数 或者变量.