tf.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. |
© 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/raw_ops/BarrierTakeMany
