TensorFlow是什么?

TensorFlow** 是一个用于数值计算的Python 库, 可以描述一幅数据计算的数据流图(data flow graph)。
TensorFlow 最初由Google大脑小组(隶属于Google机器智能研究机构)的研究员和工程师们开发出来,用于机器学习和深度神经网络方面的研究,但这个系统的通用性使其也可广泛用于其他计算领域。

数据流图如下,来自TensorFlow中文社区

TensorFlow是什么?_第1张图片
Data Flow Graph

会话 (Session):TensorFlow描述的计算流程图需要在Session中启动;Session将其与C++后端连接,为其分配计算设备(CPU 或 GPU)和提供计算方法,反复训练模型。

节点(Nodes):在图中表示数学操作,例如,数据输入(feed in)的起点或输出(push out)的终点;

线(edges):线输送节点间相互联系和不断变化的多维数据数组(即张量, tensor)。

TensorFlow 程序

TensorFlow 程序分为计算图构建和执行两个步骤;

计算图构建:除了起始和终止层外,每个操作的输入都是其它操作输出,操作的输出也是其它操作的输入。计算图构建就是将数据计算内容和过程用一个数据流图表示出来。

计算图执行:计算图构建之后,需要创建一个会话,然后才能启动图;当会话没有参数时,会话会启动默认图。

# 启动默认图.

sess = tf.Session()

#Run执行args中的数据传入,计算,以及获取最终的输出。

result = sess.run(args)

# 任务完成, 关闭会话,释放资源。

sess.close()

#”with" 代码块 来自动完成关闭动作.

with tf.Session() as sess:
  result = sess.run(args)
  print result

TensorFlow与GPU、CPU

默认情况下,TensorFlow会利用找到的第一个GPU进行运行运算,但是可以适用with...Device 来选定运算使用的CPU 或 GPU :

with tf.device("/gpu:1"):

  • "/cpu:0": 机器的 CPU.
  • "/gpu:0": 机器的第一个 GPU。
  • "/gpu:1": 机器的第二个 GPU。

交互式使用

在交互式操作中,使用一个会话启动图并不是很方便;因此可以使用 InteractiveSession 代替 Session 类和 Tensor.eval() 和 Operation.run() 方法代替 Session.run();这样就不会把计算就分放到不同的会话中来。

# 进入一个交互式 TensorFlow 会话.
import tensorflow as tf
sess = tf.InteractiveSession()

x = tf.Variable([1.0, 2.0])
a = tf.constant([3.0, 3.0])

# 使用初始化器 initializer op 的 run() 方法初始化 'x' 
x.initializer.run()

# 增加一个减法 sub op, 从 'x' 减去 'a'. 运行减法 op, 输出结果 
sub = tf.sub(x, a)
print sub.eval()
# ==> [-2. -1.]

张量

在TensorFlow计算图中,线运输的数据会不断被操作和变化;tensor 可以看作是一个 n 维的数组或列表. 一个 tensor 包含一个静态类型 rank和 一个 shape.

变量

变量维护图执行过程中的状态信息.

启动图后, 变量必须要初始化,

#将所有变量都进行初始化
init_op = tf.initialize_all_variables()  

Fetch

会话执行(Session.run)计算图时,可以将tensor的数据输出。

result = sess.run(tensor_args)

Feed

在计算图中引入tesor替代操作的输出。

学习资料来源:

TensorFlow中文社区

你可能感兴趣的:(TensorFlow是什么?)