тензорный поток:: опс:: БарьерTakeMany

#include <data_flow_ops.h>

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

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

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

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

Аргументы:

  • область: объект области .
  • handle: Дескриптор барьера.
  • num_elements: одноэлементный тензор, содержащий количество элементов, которые нужно принять.
  • компонент_типы: тип каждого компонента в значении.

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

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

Возврат:

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

Конструкторы и деструкторы

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)

Публичные атрибуты

indices
keys
operation
values

Публичные статические функции

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

Структуры

tensorflow:: ops:: BarrierTakeMany:: Attrs

Необязательные установщики атрибутов для BarrierTakeMany .

Публичные атрибуты

индексы

::tensorflow::Output indices

ключи

::tensorflow::Output keys

операция

Operation operation

ценности

::tensorflow::OutputList values

Общественные функции

БарьерTakeMany

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

БарьерTakeMany

 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
)