Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

BarrierTakeMany

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
BarrierTakeMany public final class

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

Esta operación concatena tensores de componentes de elementos completos a lo largo de la dimensión 0 para hacer un tensor de un solo componente.

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

Clases anidadas

clase BarrierTakeMany.Options Los atributos opcionales para BarrierTakeMany

Métodos públicos

estáticas BarrierTakeMany.Options
allowSmallBatch (Boolean allowSmallBatch)
estática BarrierTakeMany
crear ( Alcance alcance, operando <cadena> mango, operando <entero> numElements, List <Clase <? >> componentTypes, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación BarrierTakeMany.
Salida <Larga>
índices ()
Un tensor unidimensional de índices, con longitud num_elems.
Salida <String>
llaves ()
Un tensor unidimensional de claves, con longitud num_elements.
estáticas BarrierTakeMany.Options
timeoutMs (Long timeoutMs)
Lista < salida <? >>
valores ()
Un tensor de cualquier dimensión por componente en un elemento de barrera.
estáticas BarrierTakeMany.Options
waitForIncomplete (Boolean waitForIncomplete)

Métodos heredados

Métodos públicos

public static BarrierTakeMany.Options allowSmallBatch (booleano allowSmallBatch)

Parámetros
allowSmallBatch Permitir devolver menos de num_elements elementos si la barrera ya está cerrada.

public static BarrierTakeMany crear ( Alcance alcance, operando <cadena> mango, operando <entero> numElements, List <Clase <? >> componentTypes, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación BarrierTakeMany.

Parámetros
alcance alcance actual
resolver El mango a una barrera.
numElements Un tensor de un solo elemento que contiene la cantidad de elementos que se deben tomar.
componentTypes El tipo de cada componente en un valor.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de BarrierTakeMany

pública de salida <Larga> índices ()

Un tensor unidimensional de índices, 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).

pública de salida <String> teclas ()

Un tensor de claves unidimensional, con longitud num_elements.

estáticas pública BarrierTakeMany.Options timeoutMs (Long timeoutMs)

Parámetros
timeoutMs Si la cola está vacía, esta operación se bloqueará hasta timeout_ms milisegundos. Nota: esta opción aún no es compatible.

Lista pública < salida <? >> valores de ()

Un tensor de cualquier dimensión por componente en un elemento de barrera. Todos los valores tienen la longitud num_elements en la dimensión 0.

public static BarrierTakeMany.Options waitForIncomplete (booleano waitForIncomplete)