classe finale publique EnqueueTPUEmbeddingRaggedTensorBatch
Facilite le portage du code qui utilise tf.nn.embedding_lookup().
sample_splits[i], embedding_indices[i] et aggregation_weights[i] correspondent à la ième fonctionnalité. table_ids[i] indique quelle table d'intégration rechercher avec cette fonctionnalité.
Les tenseurs aux positions correspondantes dans deux des listes d'entrée, embedding_indices et aggregation_weights, doivent avoir la même forme, c'est-à-dire le rang 1 avec dim_size() égal au nombre total de recherches dans la table décrite par la fonctionnalité correspondante.
Classes imbriquées
| classe | Mettre en file d'attenteTPUEmbeddingRaggedTensorBatch.Options | Attributs facultatifs pour EnqueueTPUEmbeddingRaggedTensorBatch | |
Méthodes publiques
| EnqueueTPUEmbeddingRaggedTensorBatch.Options statique | combinateurs (combineurs List<String>) |
| static <T étend le nombre, U étend le nombre, V étend le nombre> EnqueueTPUEmbeddingRaggedTensorBatch | créer ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingRaggedTensorBatch. |
| EnqueueTPUEmbeddingRaggedTensorBatch.Options statique | appareilOrdinal (appareilOrdinal long) |
| EnqueueTPUEmbeddingRaggedTensorBatch.Options statique | maxSequenceLengths (Liste<Long> maxSequenceLengths) |
| EnqueueTPUEmbeddingRaggedTensorBatch.Options statique | numFeatures (Liste<Long> numFeatures) |
Méthodes héritées
Méthodes publiques
Combineurs publics statiques EnqueueTPUEmbeddingRaggedTensorBatch.Options (combineurs List<String>)
Paramètres
| combinateurs | Une liste de scalaires de chaînes, une pour chaque table d'intégration, qui spécifient comment normaliser les activations d'intégration après la sommation pondérée. Les combinateurs pris en charge sont « moyenne », « somme » ou « sqrtn ». Il n'est pas valide que la somme des poids soit égale à 0 pour « moyenne » ou que la somme des poids au carré soit égale à 0 pour « sqrtn ». Si les combinateurs ne sont pas transmis, la valeur par défaut est d'utiliser « somme » pour toutes les tables. |
|---|
public static EnqueueTPUEmbeddingRaggedTensorBatch créer ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options.. .options )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingRaggedTensorBatch.
Paramètres
| portée | portée actuelle |
|---|---|
| exemples de fractionnements | Une liste de Tensors de rang 1 spécifiant les points d'arrêt pour diviser les embedding_indices et aggregation_weights en lignes. Cela correspond à ids.row_splits dans embedding_lookup(), lorsque ids est un RaggedTensor. |
| intégrationIndices | Une liste de Tensors de rang 1, des indices dans les tables d'intégration. Cela correspond à ids.values dans embedding_lookup(), lorsque ids est un RaggedTensor. |
| agrégationPoids | Une liste de tenseurs de rang 1 contenant des poids d'agrégation par exemple de formation. Il correspond au champ de valeurs d'un RaggedTensor avec les mêmes row_splits que les ids dans embedding_lookup(), lorsque ids est un RaggedTensor. |
| modeOverride | Entrée de chaîne qui remplace le mode spécifié dans TPUEmbeddingConfiguration. Les valeurs prises en charge sont {'unspecified', 'inference', 'training', 'backward_pass_only'}. Lorsqu'il est défini sur « non spécifié », le mode défini dans TPUEmbeddingConfiguration est utilisé, sinon mode_override est utilisé. |
| ID de table | Une liste d'entiers spécifiant l'identifiant de la table d'intégration (décalage de TableDescriptor dans TPUEmbeddingConfiguration) pour rechercher l'entrée correspondante. La ième entrée est recherchée à l'aide de table_ids[i]. La taille de la liste table_ids doit être égale à celle de sample_indices, embedding_indices et aggregation_weights. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de EnqueueTPUEmbeddingRaggedTensorBatch
public static EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Paramètres
| appareilOrdinal | Le périphérique TPU à utiliser. Doit être >= 0 et inférieur au nombre de cœurs TPU dans la tâche sur laquelle le nœud est placé. |
|---|