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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法

參數

返回