tensorflow中创建多个计算图(Graph)

转自https://blog.csdn.net/dcrmg/article/details/79028032

tf中可以定义多个计算图，不同计算图上的张量和运算是相互独立的，不会共享。计算图可以用来隔离张量和计算，同时提供了管理张量和计算的机制。计算图可以通过Graph.device函数来指定运行计算的设备，为TensorFlow充分利用GPU/CPU提供了机制。

使用 g = tf.Graph()函数创建新的计算图;
在with g.as_default():语句下定义属于计算图g的张量和操作
在with tf.Session()中通过参数 graph = xxx指定当前会话所运行的计算图;
如果没有显式指定张量和操作所属的计算图，则这些张量和操作属于默认计算图;
一个图可以在多个sess中运行，一个sess也能运行多个图

创建多个计算图：

# -*- coding: utf-8 -*-)
import tensorflow as tf# 在系统默认计算图上创建张量和操作
a=tf.constant([1.0,2.0])
b=tf.constant([2.0,1.0])
result = a+b# 定义两个计算图
g1=tf.Graph()
g2=tf.Graph()# 在计算图g1中定义张量和操作
with g1.as_default():a = tf.constant([1.0, 1.0])b = tf.constant([1.0, 1.0])result1 = a + bwith g2.as_default():a = tf.constant([2.0, 2.0])b = tf.constant([2.0, 2.0])result2 = a + b# 在g1计算图上创建会话
with tf.Session(graph=g1) as sess:out = sess.run(result1)print 'with graph g1, result: {0}'.format(out)with tf.Session(graph=g2) as sess:out = sess.run(result2)print 'with graph g2, result: {0}'.format(out)# 在默认计算图上创建会话
with tf.Session(graph=tf.get_default_graph()) as sess:out = sess.run(result)print 'with graph default, result: {0}'.format(out)print g1.version  # 返回计算图中操作的个数

输出：

with graph g1, result: [ 2.  2.]
with graph g2, result: [ 4.  4.]
with graph default, result: [ 3.  3.]
3

tensorflow中创建多个计算图(Graph)相关推荐

python输入float_python – 在tensorflow中创建一个float64变量
我正在尝试进行逻辑回归,我的训练数据集来自一个numpy float64数组.我的代码看起来像, import tensorflow as tf graph = tf.Graph() with gra ...
猪年快乐之TensorFlow中实现word2vec及如何结构化TensorFlow模型
猪年快乐之TensorFlow中实现word2vec及如何结构化TensorFlow模型导语今天是2019年新年第一天,首先祝福大家猪年大吉,在新的一年里多多学习,多多锻炼,身体健康,万事如意! ...
在Tensorflow中使用深度学习构建图像标题生成器
by Cole Murray 通过科尔·默里(Cole Murray) 在Tensorflow中使用深度学习构建图像标题生成器 (Building an image caption generator ...
TensorFlow中的计算图
作者 | stephenDC 来源 | 大数据与人工智能(ID:ai-big-data) 1 什么是计算图? 一个机器学习任务的核心是模型的定义以及模型的参数求解方式,对这两者进行抽象之后,可以确定一 ...
TensorFlow中的设备管理——Device的创建与注册机制
背景 [作者:DeepLearningStack,阿里巴巴算法工程师,开源TensorFlow Contributor] 作为一款优秀的异构深度学习算法框架,TensorFlow可以在多种设备上运行算 ...
tensorflow中Tensorboard的用法
tensorflow中Tensorboard的用法下面代码定义了一个简单的用于实现常量加法和乘法的计算图. import tensorflow as tf input1 = tf.constant( ...
【AI实战】快速掌握TensorFlow（二）：计算图、会话
2019独角兽企业重金招聘Python工程师标准>>> 在前面的文章中,我们已经完成了AI基础环境的搭建(见文章:Ubuntu + Anaconda + TensorFlow + G ...
TensorFlow 中文文档介绍
介绍本章的目的是让你了解和运行 TensorFlow 在开始之前, 先看一段使用 Python API 撰写的 TensorFlow 示例代码, 对将要学习的内容有初步的印象. 这段很短的 Pyth ...
tensorflow 里metrics_深入理解TensorFlow中的tf.metrics算子
[IT168 技术]01 概述本文将深入介绍Tensorflow内置的评估指标算子,以避免出现令人头疼的问题. tf.metrics.accuracy() tf.metrics.precision( ...

tensorflow中创建多个计算图(Graph)

tensorflow中创建多个计算图(Graph)相关推荐

最新文章

热门文章