tensorflow::ops::BarrierTakeMany

#include <data_flow_ops.h>

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

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

Эта операция объединяет тензоры компонентов завершённых элементов вдоль измерения 0, чтобы сформировать один тензор компонентов.

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

Аргументы:

• 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: один тензор произвольной размерности для каждого компонента элемента барьера. Длина измерения 0 у всех значений All равна 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
)