tf.data.experimental.prefetch_to_device

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

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

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

tf.compat.v1.data.experimental.prefetch_to_device, `tf.compat.v2.data.experimental.prefetch_to_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/r1.15/api_docs/python/tf/data/experimental/prefetch_to_device