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”。