BarrierTakeMany

classe finale publique BarrierTakeMany

Prend le nombre donné d'éléments terminés d'une barrière.

Cette opération concatène les tenseurs de composants d'éléments terminés le long de la 0ème dimension pour créer un tenseur de composants unique.

Les éléments sortent de la barrière lorsqu'ils sont terminés et dans l'ordre dans lequel ils ont été placés dans la barrière. La sortie des indices fournit des informations sur le lot dans lequel chaque élément a été initialement inséré dans la barrière.

Classes imbriquées

classe BarrièreTakeMany.Options Attributs facultatifs pour BarrierTakeMany

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

statique BarrierTakeMany.Options
AllowSmallBatch (booléen AllowSmallBatch)
barrière statiqueTakeMany
créer ( Scope scope, Operand < TString > handle, Operand < TInt32 > numElements, List<Class<? extends TType >> composantTypes, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération BarrierTakeMany.
Sortie < TInt64 >
indices ()
Un tenseur d'indices unidimensionnel, de longueur num_elems.
Sortie < TString >
clés ()
Un tenseur de clés unidimensionnel, de longueur num_elements.
statique BarrierTakeMany.Options
timeoutMs (Long timeoutMs)
Liste< Sortie <?>>
valeurs ()
Un tenseur de toute dimension par composant dans un élément barrière.
statique BarrierTakeMany.Options
waitForIncomplete (booléen waitForIncomplete)

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : « BarrierTakeMany »

Méthodes publiques

public statique BarrierTakeMany.OptionsallowSmallBatch ( booléenallowSmallBatch)

Paramètres
autoriserSmallBatch Permet de renvoyer moins d'éléments num_elements si la barrière est déjà fermée.

public static BarrierTakeMany create ( Scope scope, Operand < TString > handle, Operand < TInt32 > numElements, List<Class<? extends TType >> ComponentTypes, Options... options)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération BarrierTakeMany.

Paramètres
portée portée actuelle
poignée La poignée d'une barrière.
numElements Un tenseur à élément unique contenant le nombre d'éléments à prendre.
Types de composants Le type de chaque composant dans une valeur.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de BarrierTakeMany

Sortie publique < TInt64 > indices ()

Un tenseur d'indices unidimensionnel, de longueur num_elems. Ces indices font référence au lot dans lequel les valeurs ont été placées dans la barrière (en commençant par MIN_LONG et en augmentant avec chaque BarrierInsertMany).

sortie publique < TString > clés ()

Un tenseur de clés unidimensionnel, de longueur num_elements.

public statique BarrierTakeMany.Options timeoutMs (Long timeoutMs)

Paramètres
délai d'attenteMme Si la file d'attente est vide, cette opération bloquera jusqu'à timeout_ms millisecondes. Remarque : Cette option n'est pas encore prise en charge.

public List< Sortie <?>> valeurs ()

Un tenseur de toute dimension par composant dans un élément barrière. Toutes les valeurs ont une longueur num_elements dans la 0ème dimension.

public statique BarrierTakeMany.Options waitForIncomplete (booléen waitForIncomplete)