ray.train.tensorflow.prepare_dataset_shard#

ray.train.tensorflow.prepare_dataset_shard(tf_dataset_shard: tf.data.Dataset)[源代码]#

一个覆盖 Tensorflow Dataset 默认配置的实用函数。

这应该用于通过在由 ray.train.get_dataset_shard() 返回的 ray.data.Dataset 上调用 iter_tf_batches() 创建的 TensorFlow Dataset，因为数据集已经在工作节点之间进行了分片。

参数:

tf_dataset_shard (tf.data.Dataset) – 一个 TensorFlow 数据集。

返回:

  • 自动分片已关闭

  • 预取功能已开启，自动调优已启用

返回类型:

A TensorFlow Dataset with

PublicAPI (测试版): 此API目前处于测试阶段，在成为稳定版本之前可能会发生变化。