tf.contrib.data.prefetch_to_device
Преобразование, которое предварительно загружает значения набора данных на указанное device. (устарело)
tf.contrib.data.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/contrib/data/prefetch_to_device
