Примечание: Это аварийный выход для существующих применений map_and_batch , которые не работают с функциями V2. Новые применения настоятельно не рекомендуются, и существующие должны быть мигрированы на map_and_batch, так как этот метод не будет удалён в V2.
Аргументы
map_func
Функция, отображающая вложенную структуру тензоров на другую вложенную структуру тензоров.
batch_size
Скалярное значение типа tf.int64, представляющее количество последовательных элементов данного набора данных, которые объединяются в одну партию.
num_parallel_batches
(Необязательно.) Скалярное значение типа tf.int64, представляющее количество партий, создаваемых параллельно. С одной стороны, большие значения могут помочь смягчить влияние отстающих узлов. С другой стороны, большие значения могут увеличить конкуренцию, если ЦП ограничен.
drop_remainder
(Необязательно.) Скалярное значение типа tf.bool, представляющее, следует ли отбрасывать последнюю партию в случае, если её размер меньше желаемого; по умолчанию меньшая партия не отбрасывается.
num_parallel_calls
(Необязательно.) Скалярное значение типа tf.int32, представляющее количество элементов, обрабатываемых параллельно. Если не указано, batch_size * num_parallel_batches элементов будут обработаны параллельно. Если используется значение tf.data.experimental.AUTOTUNE, то количество параллельных вызовов устанавливается динамически на основе доступного ЦП.