tensorflow 指定cpu 但是还会运行 gpu_TensorFlow中的计算图

作者 | stephenDC

来源 | 大数据与人工智能(ID:ai-big-data)

1 什么是计算图？

一个机器学习任务的核心是模型的定义以及模型的参数求解方式，对这两者进行抽象之后，可以确定一个唯一的计算逻辑，将这个逻辑用图表示，称之为计算图。计算图表现为有向无环图，定义了数据的流转方式，数据的计算方式，以及各种计算之间的相互依赖关系等。

2 计算图的基本组成

TensorFlow的计算图粒度比较细，由节点和有向边组成(后来也加入了层)。相比之下，腾讯的开源机器学习平台Angel，其计算图的粒度较粗，由层(Layer)组成。很明显，粒度越细，灵活性越好；粒度越粗，开发效率越高。用Angel手动搭建模型，层层堆叠，几行代码就够了(事实上，Angel借鉴了Caffe的方式，可以直接读取Json文件，生成深度网络)；但限制在于，只能使用官方已经实现的Layer，因此诸如R