tf.raw_ops.BarrierTakeMany

Извлекает указанное количество завершенных элементов из барьера.

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

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

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

tf.compat.v1.raw_ops.BarrierTakeMany

tf.raw_ops.BarrierTakeMany(
    handle,
    num_elements,
    component_types,
    allow_small_batch=False,
    wait_for_incomplete=False,
    timeout_ms=-1,
    name=None
)

Данная операция конкатенирует завершенные компонентные тензоры элементов по 0-му измерению, чтобы получить один компонентный тензор.

Элементы извлекаются из барьера по мере их завершения и в порядке, в котором они были помещены в барьер. Выходные индексы предоставляют информацию о партии, в которую каждый элемент изначально был вставлен в барьер.

Аргументы
`handle`	A `Tensor` типа mutable `string`. Детектор барьера.
`num_elements`	A `Tensor` типа `int32`. Одноэлементный тензор, содержащий количество извлекаемых элементов.
`component_types`	Список `tf.DTypes`, имеющий длину `>= 1`. Тип каждого компонента в значении.
`allow_small_batch`	Необязательный `bool`. По умолчанию `False`. Позволяет вернуть меньше, чем num_elements элементов, если барьер уже закрыт.
`wait_for_incomplete`	Необязательный `bool`. По умолчанию `False`.
`timeout_ms`	Необязательный `int`. По умолчанию `-1`. Если очередь пуста, эта операция будет блокироваться до тех пор, пока не истечёт время ожидания в течение timeout_ms миллисекунд. Примечание: этот параметр пока не поддерживается.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж объектов `Tensor` (индексы, ключи, значения).
`indices`	A `Tensor` типа `int64`.
`keys`	A `Tensor` типа `string`.
`values`	Список объектов `Tensor` типа `component_types`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/BarrierTakeMany