当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.data.experimental.DistributeOptions用法及代码示例


表示分布式数据处理的选项。

用法

tf.data.experimental.DistributeOptions()

属性

  • auto_shard_policy 要使用的分片类型。有关其他信息，请参阅tf.data.experimental.AutoShardPolicy。
  • num_devices 连接到此输入管道的设备数量。这将由 MultiDeviceIterator 自动设置。

您可以通过 tf.data.Options 的 experimental_distribute 属性设置数据集的分布选项；该属性是 tf.data.experimental.DistributeOptions 的一个实例。

options = tf.data.Options()
options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
dataset = dataset.with_options(options)

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.experimental.DistributeOptions。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。