📅  最后修改于: 2023-12-03 14:49:30.801000             🧑  作者: Mango
在TensorFlow中,要执行计算图,需要创建一个Session并将计算图加载到这个Session中。而启动Session的函数是tf.Session().run()
。
tf.Session().run()
函数是tensorflow中常用的一个函数,主要作用是启动会话用于运行图。
fetches
:需要观察或处理的张量或操作。可以传入一个tensor或者tensor列表,也可以是操作(operation)或操作列表。
feed_dict
:用于覆盖图中的张量的值。feed_dict是一个字典,键为张量,值为待传递给张量的值。
options
:操作执行的选项。这是一个可选参数,可以在Session.run的时候指定。
run_metadata
:该参数可以收集执行的元数据,返回一个tf.RunMetadata对象,提供关于运行时的信息。
函数的返回值根据传入的fetches
参数而定,如果fetches
传入的是单个tensor,那么函数将返回这个tensor的值;如果fetches
传入的是tensor列表,那么函数将返回一个列表,其中包含每个tensor的值;如果fetches
传入的是operation,那么函数将返回None
。
下面是一个简单的例子:
import tensorflow as tf
# 创建计算图
a = tf.constant(1)
b = tf.constant(2)
c = tf.add(a, b)
# 创建Session并执行
with tf.Session() as sess:
result = sess.run(c)
print(result)
输出:
3
参考资料: