XlaRecvTPUEmbeddingActivations

classe finale publique XlaRecvTPUEmbeddingActivations

Une opération qui reçoit les activations d’intégration sur le TPU.

Le système TPU effectue les recherches et les agrégations d'intégration. Les résultats de ces agrégations sont visibles par le graphique Tensorflow en tant que sorties d'une opération XlaRecvTPUEmbeddingActivations. Cette opération renvoie une liste contenant un Tensor d'activations par table spécifiée dans le modèle.

Méthodes publiques

XlaRecvTPUEmbeddingActivations statiques	créer ( Scope scope, Operand <?> deduplicationData, Long numTables, String config) Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaRecvTPUEmbeddingActivations.
Itérateur< Opérande <Float>>	itérateur ()
Liste< Sortie <Float>>	les sorties () Une TensorList d'activations d'intégration contenant un Tensor par table d'intégration dans le modèle.

Méthodes héritées

De la classe org.tensorflow.op.PrimitiveOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie l' `Operation` sous-jacente
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface java.lang.Iterable

vide	forEach (Consommateur <? super T> arg0)
Itérateur abstrait< Opérande <Float>>	itérateur ()
Spliterator< Opérande <Float>>	séparateur ()

Méthodes publiques

public static XlaRecvTPUEmbeddingActivations créer ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)

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

Paramètres

portée	portée actuelle
données de déduplication	Un Tensor avec type=DT_VARIANT contenant les données de déduplication. Le tenseur est un tuple imbriqué XLA contenant N éléments (où N est le rapport entre le nombre d'intégrations et le nombre de cœurs de tenseur par puce TPU). Chaque élément du tuple imbriqué est un tuple de tenseurs de rang 1. Chaque tenseur contient soit des indices (DT_UINT32) pour la recherche d'intégration sur le TensorCore, soit des poids (DT_FLOAT) à appliquer à la sortie de l'opération de recherche d'intégration.
numTables	Le nombre de tenseurs d'activation de sortie. Si le descripteur de fonctionnalité est présent dans la configuration d'intégration tpu, il est égal au nombre de fonctionnalités, autrement égal au nombre de tables d'intégration dans le modèle.
configuration	Protocole TPUEmbeddingConfiguration sérialisé.

Retour

une nouvelle instance de XlaRecvTPUEmbeddingActivations

public Iterator< Opérande <Float>> itérateur ()

public List< Output <Float>> sorties ()

Une TensorList d'activations d'intégration contenant un Tensor par table d'intégration dans le modèle.