Skip to content

第一个hello tensorflow

In [1]: import tensorflow

In [2]: hello = tensorflow.constant('hello tenforflow')

In [3]: with tensorflow.Session() as session:
   ...:     session.run(hello)
   ...:
2018-12-17 16:31:49.667318: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

In [4]: with tensorflow.Session() as session:
   ...:     print(session.run(hello))
   ...:
   ...:
b'hello tenforflow'

总结

这里是第一个tensorflow程序,在分析之前当然是需要对tensorflow进行一些介绍。
这个名字可以分解为两个单词理解:Tensor(张量) + Flow(数据流)
在希望使用Tensorflow做一些事情时,一般需要三个步骤:
1. 创建Tensor
2. 添加Operation(op),由op输入一个tensor,输出另外一个tensor
3. 执行计算

Tensorflow中有一个图的概念,op会作为节点添加至图中。在添加某些op时,不会立即执行。Tensorflow会等待所有op添加结束后优化这张图,以便决定如何计算。

比如上面的代码中的hello便是一个Constant op。构造函数返回的值表示Constant op的输出。
Session提供了op的计算环境。使用会话管理可以很优雅地调用Session。之后执行这张图即可获得结果。