tensorflow::ops::BarrierTakeMany

#include <data_flow_ops.h>

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

Краткое описание

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

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

Аргументы:

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

Необязательные атрибуты (см. Attrs):

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

Возвращаемое значение:

• Output indices: Одномерный тензор индексов длиной num_elems. Эти индексы ссылаются на партию, в которой значения были помещены в барьер (начиная с MIN_LONG и увеличиваясь при каждом вызове BarrierInsertMany).
• Output keys: Одномерный тензор ключей длиной 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
)