Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse().
embedding_indices[i] et aggregation_weights[i] correspondent à la ième 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'attenteTPUEmbeddingArbitraryTensorBatch.Options | Attributs facultatifs pour EnqueueTPUEmbeddingArbitraryTensorBatch | |
Méthodes publiques
| EnqueueTPUEmbeddingArbitraryTensorBatch.Options statique | combinateurs (combineurs List<String>) |
| static <T étend le nombre, U étend le nombre, V étend le nombre> EnqueueTPUEmbeddingArbitraryTensorBatch | créer ( Scope scope, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingArbitraryTensorBatch. |
| EnqueueTPUEmbeddingArbitraryTensorBatch.Options statique | appareilOrdinal (appareilOrdinal long) |
Méthodes héritées
Méthodes publiques
Combineurs publics statiques EnqueueTPUEmbeddingArbitraryTensorBatch.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 EnqueueTPUEmbeddingArbitraryTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingArbitraryTensorBatch.
Paramètres
| portée | portée actuelle |
|---|---|
| sampleIndicesOrRowSplits | Une liste de Tensors de rang 2 spécifiant l'exemple de formation auquel appartiennent les valeurs embedding_indices et aggregation_weights correspondantes. Si la taille de sa première dimension est 0, nous supposons que chaque embedding_indices appartient à un échantillon différent. Int32 et int64 sont autorisés et seront convertis en int32 en interne. Ou une liste de Tensors de rang 1 spécifiant les divisions de lignes pour diviser les embedding_indices et les aggregation_weights en lignes. Cela correspond à ids.row_splits dans embedding_lookup(), lorsque ids est un RaggedTensor. Lors de la mise en file d'attente du tenseur irrégulier ND, seule la dernière dimension peut être irrégulière. les divisions de lignes sont un tenseur dense 1-D. Lorsqu'il est vide, nous supposons qu'un tenseur dense est transmis à l'op. Int32 et int64 sont autorisés et seront convertis en int32 en interne. |
| intégrationIndices | Une liste de Tensors de rang 1, des indices dans les tables d'intégration. Int32 et int64 sont autorisés et seront convertis en int32 en interne. |
| agrégationPoids | Une liste de tenseurs de rang 1 contenant des poids d'agrégation par exemple de formation. float32 et float64 sont autorisés et seront convertis en float32 en interne. |
| 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é. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de EnqueueTPUEmbeddingArbitraryTensorBatch
public static EnqueueTPUEmbeddingArbitraryTensorBatch.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é. |
|---|