Python tf.distribute.OneDeviceStrategy.experimental_distribute_dataset用法及代码示例

用法

experimental_distribute_dataset(
    dataset, options=None
)

分发通过数据集提供的 tf.data.Dataset 实例。

在这种情况下，只有一个设备，所以这只是输入数据集的一个薄包装器。但是，它会将输入数据预取到指定的设备。返回的分布式数据集可以像常规数据集一样进行迭代。

注意：目前，用户无法向分布式数据集添加更多转换。

例子：

strategy = tf.distribute.OneDeviceStrategy()
dataset = tf.data.Dataset.range(10).batch(2)
dist_dataset = strategy.experimental_distribute_dataset(dataset)
for x in dist_dataset:
  print(x)  # [0, 1], [2, 3],...

Args：数据集：tf.data.Dataset 要预取到设备。选项：tf.distribute.InputOptions 用于控制有关此数据集如何分布的选项。返回：调用者可以迭代的“分布式Dataset”。