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