tensorflow图的基本操作

最新推荐文章于 2025-09-05 01:35:21 发布

原创最新推荐文章于 2025-09-05 01:35:21 发布 · 798 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#图的基本操作

AI 同时被 2 个专栏收录

20 篇文章

订阅专栏

tensorflow

15 篇文章

订阅专栏

本文介绍了TensorFlow中图的基本操作，包括如何创建图、获取当前默认图及重置默认图等，并通过实例演示了这些操作的具体用法。此外，还详细讲解了如何使用TensorFlow API获取张量、操作节点及元素列表。

TensorFlow 会默认建立一个图, 还可以手动建立图图的三个基本操作：

tf.Graph() # 建立图
tf.get_default_graph() # 获取图
tf.reset_default_graph() # 重置图

下面是一个简单的例子：

import tensorflow  as tf

# 默认图中的常量

default_const = tf.constant(1.)

# 新建一个图

new_graph = tf.Graph()
with new_graph.as_default():
    # 在新建的图中设置常量
    in_new_graph = tf.constant(2., name='in_new_graph')
    print(in_new_graph.graph)    # 新建图
    print(new_graph)
    print(default_const.graph)    # 默认的图

# 获得默认的图

default_graph = tf.get_default_graph()
print(default_graph)

# 重置图

tf.reset_default_graph()
new_graph1 = tf.get_default_graph()
print(new_graph1)

# 上面测试代码的输出结果

# <tensorflow.python.framework.ops.Graph object at 0x112ce3eb8>
# <tensorflow.python.framework.ops.Graph object at 0x112ce3eb8>
# <tensorflow.python.framework.ops.Graph object at 0x103bfb048>
# <tensorflow.python.framework.ops.Graph object at 0x103bfb048>
# <tensorflow.python.framework.ops.Graph object at 0x112ce3fd0>

从上面的结果中可以看出，默认图，新建图，和重置后的图的id是不同的。

# 获取张量 get_tensor_by_name()

print(in_new_graph.name)
tensor = new_graph.get_tensor_by_name('in_new_graph:0')
print(tensor)

# 获取操作节点 get_operation_by_name()

op_ = new_graph.get_operation_by_name('in_new_graph')
print(op_)

# name: "in_new_graph"
# op: "Const"
# attr {
#   key: "dtype"
#   value {
#     type: DT_FLOAT
#   }
# }
# attr {
#   key: "value"
#   value {
#     tensor {
#       dtype: DT_FLOAT
#       tensor_shape {
#       }
#       float_val: 2.0
#     }
#   }
# }

# 获取元素列表 get_operations()

ng = new_graph.get_operations()
print(ng)    # [<tf.Operation 'in_new_graph' type=Const>]

# 获取对象 tf.Graph.as_graph_element(obj, allow_tensor=True, allow_operation=True)
# 传入一个对象返回一个张量或是一个OP

age = new_graph.as_graph_element(in_new_graph)
print(age)    # Tensor("in_new_graph:0", shape=(), dtype=float32)