3.1TensorFlow计算模型--计算图

TensorFlow中所有的计算都会被转化为计算图上的节点

3.1.1计算图的概念

TensorFlow = Tensor(张量:理解为多维数组) + Flow(流)

Tensor表明数据结构

Flow体现计算模型,表达张量之间通过计算相互转化的过程

3.1.2计算图的使用

TensorFlow程序一般分为两个阶段:1.定义计算 2.执行计算

定义计算的样例:

import tensorflow as tf:TensorFlow会自动将定义的计算转化为计算图上的节点.在TensorFlow程序中,系统会默认维护一个默认的计算图,通过tf.get_default_graph函数可以获取当前默认的计算图

除了使用默认的计算图,TensorFlow支持通过tf.Graph函数来生成新的计算图,不同计算图上的张量和运算都不会共享,以下代码示意了如何在不同计算图上定义和使用变量

TensorFlow中的计算图不仅可以用来隔离张量和计算,他还提供了管理张量和计算的机制.

计算图可以通过tf.Graph.device函数来指定运行计算的设备,这位TensorFlow使用GPU提供了机制.

猜你喜欢

转载自blog.csdn.net/wfx18765903641/article/details/84919031
今日推荐