tensorflow:: ops:: BarrierTakeMany

Toma el número dado de elementos completados de una barrera.

Resumen

Esta operación concatena tensores de componentes de elementos completos a lo largo de la dimensión 0 para formar un único tensor de componentes.

Los elementos salen de la barrera cuando están completos y en el orden en que fueron introducidos en ella. Los índices de salida proporcionan información sobre el lote en el que cada elemento fue insertado originalmente en la barrera.

Argumentos:

• ámbito: Un objeto de ámbito
• asa: El asa de una barrera.
• num_elements: Un tensor de un solo elemento que contiene el número de elementos a tomar.
• component_types: El tipo de cada componente en un valor.

Atributos opcionales (ver Attrs ):

• allow_small_batch: Permite devolver menos de num_elements elementos si la barrera ya está cerrada.
• timeout_ms: Si la cola está vacía, esta operación se bloqueará durante un máximo de timeout_ms milisegundos. Nota: Esta opción aún no es compatible.

Devoluciones:

• Índices Output : Un tensor unidimensional de índices, con una longitud de num_elems. Estos índices se refieren al lote en el que se colocaron los valores en la barrera (comenzando con MIN_LONG y aumentando con cada BarrierInsertMany ).
• Claves Output : Un tensor unidimensional de claves, con una longitud de num_elements.
• Valores OutputList : Un tensor de cualquier dimensión por componente en un elemento de barrera. Todos los valores tienen una longitud de num_elements en la dimensión 0.

Atributos públicos

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