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.3/api_docs/python/tf/data/experimental/prefetch_to_device