tensorflow::ops::BarrierTakeMany

#include <data_flow_ops.h>

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

Summary

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

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

Аргументы:

• 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
)