tensorflow.python.client.session.BaseSession.run
功能描述
TensorFlow执行计算图的方法。
函数原型
def run(self, fetches, feed_dict=None, options=None, run_metadata=None)
参数说明
参数 |
类型 |
可选/必选 |
说明 |
|---|---|---|---|
fetches |
str, tf.Operation, tf.Variable, tf.Tensor, tf.sparse.SparseTensor, list, tuple, dict |
必选 |
运行操作或者获取其中的Tensor |
feed_dict |
tf.Variable, tf.Tensor, tf.sparse.SparseTensor, list, tuple, dict |
可选 |
覆盖图中Tensor的值 |
options |
tf.compat.v1.RunOptions |
可选 |
控制特定步骤的行为 |
run_metadata |
tf.compat.v1.RunMetadata |
可选 |
在特定步骤时,收集非张量输出 |
返回值说明
- 成功:如果fetches是单个元素,则为单个值;如果fetches是list,则为值list;如果fetches是dict,则为具有相同键的dict。
- 失败:抛出异常。
使用示例
以下仅提供使用流程的示例。
# 1、导入需要的库
import tensorflow as tf
from mx_rec.util.initialize import init
# 2、构建计算图
# ...
# 3、调用接口训练
with tf.compat.v1.Session() as sess:
sess.run([train_ops]) #train_ops为构建计算图中构建的训练算子
父主题: TensorFlow相关接口