TPUEmbeddingActivations

classe finale publique TPUEmbeddingActivations

Une opération permettant de différencier les TPU Embeddings.

Cette opération renvoie simplement sa première entrée, qui est supposée avoir été découpée à partir des Tensors renvoyés par TPUEmbeddingDequeueActivations. La présence de cette opération, et son premier argument étant une variable entraînable, permet la différenciation automatique des graphiques contenant des intégrations via les bibliothèques TPU Embedding Python.

Méthodes publiques

Sortie <Flottant>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
TPUEmbeddingActivations statiques
créer ( Portée de la portée , Opérande <Float> embeddingVariable, Opérande <Float> slicedActivations, Long tableId, Long lookupId)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUEmbeddingActivations.
Sortie <Flottant>
sortir ()

Méthodes héritées

Méthodes publiques

sortie publique <Float> asOutput ()

Renvoie le handle symbolique d'un tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static TPUEmbeddingActivations create ( Scope scope, Operand <Float> embeddingVariable, Operand <Float> slicedActivations, Long tableId, Long lookupId)

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

Paramètres
portée portée actuelle
variable d'intégration Une variable entraînable, permettant aux optimiseurs de trouver cette opération.
Activations tranchées Le Tensor d'activations d'intégration doit revenir.
ID de table L'identifiant de la table dans la configuration de la couche d'intégration à partir de laquelle ces activations ont été calculées.
ID de recherche Identifiant de l'ensemble des indices d'incorporation qui ont produit ces activations.
Retour
  • une nouvelle instance de TPUEmbeddingActivations

Sortie publique Sortie <Float> ()