tf.ConfigProto()
现在记录一下我们经常会用到的tf.ConfigProto()函数的功能和使用
一、功能
tf.ConfigProto()
函数用在创建session的时候对session进行参数配置。
首先需要了解一下session,这样才可以更好的使用该函数。
二、session
session是tensorflow框架中的一个重要机制,负责分配和管理资源。Tensorflow依赖于一个高效的C++后端来进行计算,与后端的这个连接叫做session。
在tensorflow中使用图(graph)表示计算任务,为了进行计算,图必须在会话(session)里启动,会话能够将图的节点(op)分发到CPU或GPU设备上,同时执行op方法,这些方法执行后将产生的数据(tensor)返回。会话提供了一个run方法,用于执行计算图整体或部分节点。
简单的说就是使用会话Session
启动图,调用Session.run()
执行操作,另外需通过上下文管理机制with
管理该会话,会话结束后自动释放资源,格式如下所示:
with tf.Session() as sess:
sess.run(...)
备注:
在tensor