tensorflow::ops::BarrierTakeMany

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

Краткое содержание

Эта операция объединяет тензоры компонент завершенных элементов вдоль нулевого измерения, образуя единый тензор компонент.

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

Аргументы:

• scope: Объект Scope
• ручка: ручка ограждения.
• num_elements: Тензор, содержащий один элемент и количество элементов, которые необходимо взять.
• component_types: Тип каждого компонента в значении.

Дополнительные атрибуты (см. Attrs ):

• allow_small_batch: Разрешить возврат менее num_elements элементов, если барьер уже закрыт.
• timeout_ms: Если очередь пуста, эта операция будет заблокирована на время до timeout_ms миллисекунд. Примечание: Эта опция пока не поддерживается.

Возвращает:

• Output индексы: одномерный тензор индексов длиной num_elems. Эти индексы указывают на пакет, в котором значения были помещены в барьер (начиная с MIN_LONG и увеличиваясь с каждым BarrierInsertMany ).
• Output ключи: одномерный тензор ключей длиной num_elements.
• OutputList values: Один многомерный тензор на каждый компонент в элементе барьера. Все значения имеют длину num_elements в нулевом измерении.

Общедоступные атрибуты

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)