Python tf.Graph.colocate_with用法及代码示例

用法

@tf_contextlib.contextmanager
colocate_with(
    op, ignore_existing=False
)

返回一个上下文管理器，它指定一个要与之共置的操作。

注意：此函数不供公众使用，仅供内部 Library 使用。

例如：

a = tf.Variable([1.0])
with g.colocate_with(a):
  b = tf.constant(1.0)
  c = tf.add(a, b)

b 和 c 将始终与 a 位于同一位置，无论最终将 a 放置在何处。

注意：使用托管范围会重置任何现有的设备约束。

如果 op 是 None 则 ignore_existing 必须是 True 并且新范围会重置所有托管和设备约束。