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

将数据集值预取到给定 device 的转换。

用法

tf.data.experimental.prefetch_to_device(
    device, buffer_size=None
)

参数

device 一个字符串。将预取元素的设备的名称。
buffer_size (可选。)要在 device 上缓冲的元素数。默认为自动选择的值。

一个 Dataset 转换函数，可以传递给 tf.data.Dataset.apply 。

注意：尽管转换会创建 tf.data.Dataset ，但转换必须是输入管道中的最终 Dataset。

例如，

>>> dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
>>> dataset = dataset.apply(tf.data.experimental.prefetch_to_device("/cpu:0"))
>>> for element in dataset:
...   print(f'Tensor {element} is on device {element.device}')
Tensor 1 is on device /job:localhost/replica:0/task:0/device:CPU:0
Tensor 2 is on device /job:localhost/replica:0/task:0/device:CPU:0
Tensor 3 is on device /job:localhost/replica:0/task:0/device:CPU:0

相关用法

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

用法

参数

返回