classe finale publique EnqueueTPUEmbeddingSparseTensorBatch
Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse().
sample_indices[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 les trois listes d'entrée (sample_indices, 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'attenteTPUEmbeddingSparseTensorBatch.Options | Attributs facultatifs pour EnqueueTPUEmbeddingSparseTensorBatch | |
Méthodes publiques
| EnqueueTPUEmbeddingSparseTensorBatch.Options statique | combinateurs (combineurs List<String>) |
| static <T étend le nombre, U étend le nombre, V étend le nombre> EnqueueTPUEmbeddingSparseTensorBatch | créer ( Scope scope, Iterable< Operand <T>> sampleIndices, 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 EnqueueTPUEmbeddingSparseTensorBatch. |
| EnqueueTPUEmbeddingSparseTensorBatch.Options statique | appareilOrdinal (appareilOrdinal long) |
| EnqueueTPUEmbeddingSparseTensorBatch.Options statique | maxSequenceLengths (Liste<Long> maxSequenceLengths) |
| EnqueueTPUEmbeddingSparseTensorBatch.Options statique | numFeatures (Liste<Long> numFeatures) |
Méthodes héritées
Méthodes publiques
Combineurs publics statiques EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, 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 EnqueueTPUEmbeddingSparseTensorBatch.
Paramètres
| portée | portée actuelle |
|---|---|
| exemplesIndices | Une liste de Tensors de rang 1 spécifiant l'exemple de formation auquel appartiennent les valeurs embedding_indices et aggregation_weights correspondantes. Cela correspond à sp_ids.indices[:,0] dans embedding_lookup_sparse(). |
| intégrationIndices | Une liste de Tensors de rang 1, des indices dans les tables d'intégration. Cela correspond à sp_ids.values dans embedding_lookup_sparse(). |
| agrégationPoids | Une liste de tenseurs de rang 1 contenant des poids d'agrégation par exemple de formation. Cela correspond à sp_weights.values dans embedding_lookup_sparse(). |
| 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 EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.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é. |
|---|