当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.compat.v1.Session.partial_run用法及代码示例


用法

partial_run(
    handle, fetches, feed_dict=None
)

参数

  • handle 一系列部分运行的句柄。
  • fetches 单个图形元素、图形元素列表或其值为图形元素或图形元素列表的字典(请参阅run 的文档)。
  • feed_dict 将图形元素映射到值的字典(如上所述)。

返回

  • 如果 fetches 是单个图形元素，则为单个值；如果 fetches 是列表，则为值列表；如果是字典，则为与 fetches 具有相同键的字典(请参阅run 的文档)。

抛出

继续执行更多的提要和获取。

这是实验性的，可能会发生变化。

要使用部分执行，用户首先调用 partial_run_setup() ，然后调用 partial_run() 序列。 partial_run_setup 指定将在后续 partial_run 调用中使用的提要和提取列表。

可选的 feed_dict 参数允许调用者覆盖图中张量的值。有关详细信息，请参阅run()。

下面是一个简单的例子：

a = array_ops.placeholder(dtypes.float32, shape=[])
b = array_ops.placeholder(dtypes.float32, shape=[])
c = array_ops.placeholder(dtypes.float32, shape=[])
r1 = math_ops.add(a, b)
r2 = math_ops.multiply(r1, c)

h = sess.partial_run_setup([r1, r2], [a, b, c])
res = sess.partial_run(h, r1, feed_dict={a:1, b:2})
res = sess.partial_run(h, r2, feed_dict={c:res})

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.Session.partial_run。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。