tf.data.experimental.parallel_interleave

Параллельная версия преобразования Dataset.interleave(). (устарело)

Просмотр псевдонимов

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

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

tf.compat.v1.data.experimental.parallel_interleave

tf.data.experimental.parallel_interleave(
    map_func, cycle_length, block_length=1, sloppy=False,
    buffer_output_elements=None, prefetch_input_elements=None
)

parallel_interleave() отображает map_func по всем входным данным для создания вложенных наборов данных и выводит их переплетённые элементы. В отличие от tf.data.Dataset.interleave, она получает элементы из cycle_length вложенных наборов данных параллельно, что повышает производительность, особенно при наличии отстающих элементов. Кроме того, аргумент sloppy можно использовать для повышения производительности, ослабив требование о том, что результаты должны генерироваться в детерминированном порядке, и позволяя реализации пропускать вложенные наборы данных, элементы которых не доступны при запросе.

Пример использования:

# Preprocess 4 files concurrently.
filenames = tf.data.Dataset.list_files("/path/to/data/train*.tfrecords")
dataset = filenames.apply(
    tf.data.experimental.parallel_interleave(
        lambda filename: tf.data.TFRecordDataset(filename),
        cycle_length=4))

Аргументы
`map_func`	Функция, отображающая вложенную структуру тензоров в `Dataset`.
`cycle_length`	Количество входных `Dataset` для параллельного переплетения.
`block_length`	Количество последовательных элементов, которые нужно извлечь из входного `Dataset` перед переходом к следующему входному `Dataset`.
`sloppy`	Логическое значение, определяющее, следует ли пожертвовать детерминизмом ради производительности, разрешив генерировать элементы в произвольном порядке. Если `sloppy` равно `None`, используется параметр набора данных `tf.data.Options.experimental_deterministic` (`True` по умолчанию), чтобы определить, следует ли соблюдать детерминированный порядок.
`buffer_output_elements`	Количество элементов, которые каждый итерируемый объект должен буферизовать (аналогично преобразованию `.prefetch()` для каждого переплетённого итерируемого объекта).
`prefetch_input_elements`	Количество входных элементов для преобразования в итераторы до того, как они понадобятся для переплетения.

Возвращаемое значение
Функция преобразования `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/parallel_interleave