tf.compat.v1.tpu.batch_parallel

Разделяет computation по размеру пакетной размерности для параллельного выполнения.

tf.compat.v1.tpu.batch_parallel(
    computation, inputs=None, num_shards=1, infeed_queue=None,
    device_assignment=None, name=None
)

Обёртка для удобства работы со shard().

inputs должен быть списком тензоров или None (эквивалентно пустому списку). Каждый вход разбивается на num_shards части по 0-й размерности, и вычисление применяется к каждой части параллельно.

Тензоры транслируются на все части, если они лексически захвачены computation. Например:

x = tf.constant(7) def computation(): return x + 3 ... = shard(computation, ...)

Выходы всех частей конкатенируются обратно вместе по их 0-й размерности.

Входы и выходы вычисления должны быть тензорами с рангом не менее 1.

Аргументы
`computation`	Функция Python, которая создаёт вычисление для применения к каждой части входных данных.
`inputs`	Список входных тензоров или None (эквивалентно пустому списку). 0-я размерность каждого тензора должна быть кратна `num_shards`.
`num_shards`	Количество частей.
`infeed_queue`	Если не `None`, `InfeedQueue` из которого добавить кортеж аргументов как входов в `computation`.
`device_assignment`	Если не `None`, `DeviceAssignment` описание сопоставления логических ядер в вычислении с физическими ядрами в топологии TPU. Используется значение по умолчанию для назначения устройства, если `None`. `DeviceAssignment` можно опустить, если каждая часть вычисления использует только одно ядро, и либо есть только одна часть, либо число частей равно числу ядер в системе TPU.
`name`	(Устарело) Не выполняет никаких действий.

Возвращает
Список выходных тензоров.

Возможные исключения
`ValueError`	Если `num_shards <= 0`

© 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.3/api_docs/python/tf/compat/v1/tpu/batch_parallel