【tensorflow】【python】tf.Session.run()

资源存储库

已于 2024-01-14 17:03:39 修改

阅读量520

点赞数 7

文章标签： python

于 2024-01-14 11:26:46 首次发布

本文链接：https://blog.csdn.net/wq6qeg88/article/details/135580998

版权

本文详细解释了TensorFlow中的Session.run()方法，包括其作用——获取fetches操作的输出，以及如何使用feed_dict来传递图元素值。通过示例展示了如何在不同场景中运用这两个关键概念。

摘要由CSDN通过智能技术生成

import tensorflow as tf
SESS = tf.Session()
SESS.run()
tensorflow.Session.run()
tf.Session.run()

含义

取回（Fetch）操作的输出内容, 可在使用 Session 对象的 run()调用执行图时，传入一些 tensor。这些 tensor 会帮助取回结果。

语法

run(fetches,   
    feed_dict=None)

参数

fetches：单个图元素，图元素列表或字典，字典元素其值是图元素或图元素列表。可以是单个图元素(single graph element)，也可以是任意嵌套的列表list，元组tuple，名称元组namedtuple，字典dict或包含图元素的OrderedDict。

feed_dict：将图元素映射到值的字典。{图元素：对应数值}。可选参数 feed_dict允许调用者替换图中张量的值(the value of tensors in the graph)。

返回值

如果fetches是单个图元素则返回单个值，如果fetches是列表则返回值列表，如果fetches是字典则返回具有相同键的字典。

示例1

feed_dict = {
     self.AC.s: buffer_s,
     self.AC.a_his: buffer_a,
     self.AC.v_target: buffer_v_target,
}
SESS.run([self.update_a_op, self.update_c_op], feed_dict)

在这里插入图片描述

fetches：[self.update_a_op, self.update_c_op]
fetche1：self.update_a_op
fetche2：self.update_c_op
feed_dict = {
self.AC.s: buffer_s,
self.AC.a_his: buffer_a,
self.AC.v_target: buffer_v_target,
}

示例2

prob_weights = SESS.run(self.a_prob, feed_dict={self.s: s[np.newaxis, :]})

fetches：self.a_prob
feed_dict = {self.s: s[np.newaxis, :]}
在这里插入图片描述

SESS.run([self.update_a_op, self.update_c_op], feed_dict)
SESS.run([self.pull_a_params_op, self.pull_c_params_op])
prob_weights = SESS.run(self.a_prob, feed_dict={self.s: s[np.newaxis, :]})