tf.data.experimental.prefetch_to_device

Преобразование, которое предзагружает значения набора данных на заданный device.

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

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

Примечание: Несмотря на то, что преобразование создает tf.data.Dataset, преобразование должно быть последним Dataset в конвейере обработки.

Аргументы
`device`	Строка. Имя устройства, на которое будут предзагружены элементы.
`buffer_size`	(Необязательно.) Количество элементов для буферизации на `device`. По умолчанию используется автоматически выбранное значение.

Возвращаемое значение
Функция преобразования `Dataset`, которую можно передать в `tf.data.Dataset.apply`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/data/experimental/prefetch_to_device