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 並且新範圍會重置所有托管和設備約束。