flujo tensor:: operaciones:: barreratomarmuchos

Toma el número dado de elementos completos 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 tensor de un solo componente.

Los elementos salen de la barrera cuando están completos y en el orden en que fueron colocados en la barrera. La salida de los índices proporciona información sobre el lote en el que cada elemento se insertó originalmente en la barrera.

Argumentos:

• alcance: un objeto de alcance
• mango: El mango de una barrera.
• num_elements: un tensor de un solo elemento que contiene el número de elementos a tomar.
• type_types: el tipo de cada componente en un valor.

Atributos opcionales (ver Attrs ):

• enable_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á por hasta timeout_ms milisegundos. Nota: Esta opción aún no es compatible.

Devoluciones:

• Índices Output : un tensor de índices unidimensional, con longitud 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 de claves unidimensional, con longitud num_elements.
• Valores OutputList : un tensor de cualquier dimensión por componente en un elemento de barrera. Todos los valores tienen longitud 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
)