Python tf.distribute.experimental.coordinator.ClusterCoordinator.fetch用法及代码示例

用法

fetch(
    val
)

阻止调用以从远程值中获取结果。

这是RemoteValue 结构的tf.distribute.experimental.coordinator.RemoteValue.fetch 的包装器；它返回RemoteValue s的执行结果。如果尚未准备好，请在阻止调用者的同时等待他们。

例子：

strategy = ...
coordinator = tf.distribute.experimental.coordinator.ClusterCoordinator(
    strategy)

def dataset_fn():
  return tf.data.Dataset.from_tensor_slices([1, 1, 1])

with strategy.scope():
  v = tf.Variable(initial_value=0)

@tf.function
def worker_fn(iterator):
  def replica_fn(x):
    v.assign_add(x)
    return v.read_value()
  return strategy.run(replica_fn, args=(next(iterator),))

distributed_dataset = coordinator.create_per_worker_dataset(dataset_fn)
distributed_iterator = iter(distributed_dataset)
result = coordinator.schedule(worker_fn, args=(distributed_iterator,))
assert coordinator.fetch(result) == 1