flujo tensor:: operaciones:: barreratomarmuchos
#include <data_flow_ops.h>
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 componente único.
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.
Constructores y destructores | |
---|---|
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) |
Atributos públicos | |
---|---|
indices | |
keys | |
operation | |
values |
Funciones estáticas públicas | |
---|---|
AllowSmallBatch (bool x) | |
TimeoutMs (int64 x) | |
WaitForIncomplete (bool x) |
estructuras | |
---|---|
tensorflow:: operaciones:: BarrierTakeMany:: Atributos | Configuradores de atributos opcionales para BarrierTakeMany . |
Atributos públicos
índices
::tensorflow::Output indices
llaves
::tensorflow::Output keys
operación
Operation operation
valores
::tensorflow::OutputList values
Funciones públicas
barreratomarmuchos
BarrierTakeMany( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_elements, const DataTypeSlice & component_types )
barreratomarmuchos
BarrierTakeMany( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_elements, const DataTypeSlice & component_types, const BarrierTakeMany::Attrs & attrs )
Funciones estáticas públicas
Permitir lote pequeño
Attrs AllowSmallBatch( bool x )
Tiempo de espera
Attrs TimeoutMs( int64 x )
Esperar por incompleto
Attrs WaitForIncomplete( bool x )