EnqueueTPUEmbeddingSparseBatch

public final class EnqueueTPUEmbeddingSparseBatch

Un op che accoda gli indici di input TPUEmbedding da uno SparseTensor.

Questa operazione facilita il porting del codice che utilizza embedding_lookup_sparse (), sebbene sia necessaria una pre-elaborazione Python degli argomenti SparseTensor per embedding_lookup_sparse () per produrre gli argomenti per questa operazione, poiché è consentita solo una singola operazione EnqueueTPUEmbeddingSparseBatch per fase di addestramento.

I tensori nelle posizioni corrispondenti nelle tre liste di input devono avere la stessa forma, cioè rango 1 con dim_size () uguale al numero totale di ricerche nella tabella descritta dal corrispondente table_id.

Classi annidate

classe EnqueueTPUEmbeddingSparseBatch.Options Attributi facoltativi per EnqueueTPUEmbeddingSparseBatch

Metodi pubblici

static EnqueueTPUEmbeddingSparseBatch.Options
combinatori (List <String> combiners)
static <T estende il numero, U estende il numero, V estende il numero> EnqueueTPUEmbeddingSparseBatch
create ( Scope scope, Iterable < Operand <T>> sampleIndices, Iterable < Operand <U>> embeddingIndices, Iterable < Operand <V>> aggregationWeights, Operand <String> modeOverride, Opzioni ... )
Metodo Factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingSparseBatch.
static EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (dispositivo lungoOrdinale)

Metodi ereditati

Metodi pubblici

public static EnqueueTPUEmbeddingSparseBatch.Options combiners (List <String> combiners)

Parametri
combinatori Un elenco di scalari di stringa, uno per ciascuna tabella di incorporamento che specifica come normalizzare le attivazioni di incorporamento dopo la somma ponderata. I combinatori supportati sono "mean", "sum" o "sqrtn". Non è valido che la somma dei pesi sia 0 per "media" o la somma dei pesi al quadrato sia 0 per "sqrtn". Se i combinatori non vengono passati, l'impostazione predefinita è utilizzare "sum" per tutte le tabelle.

public static EnqueueTPUEmbeddingSparseBatch create ( Scope scope, Iterable < Operand <T>> sampleIndices, Iterable < Operand <U>> embeddingIndices, Iterable < Operand <V>> aggregationWeights, Operand <String> modeOverride, Opzioni ... )

Metodo Factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingSparseBatch.

Parametri
scopo ambito attuale
sampleIndices Un elenco di tensori di rango 1 che specifica l'esempio di addestramento e la funzione a cui appartengono i valori embedding_indices e aggregation_weights corrispondenti. sample_indices [i] deve essere uguale a b * nf + f, dove nf è il numero di caratteristiche dalla tabella corrispondente, f è in [0, nf) e b è in [0, batch size).
embeddingIndices Un elenco di tensori di rango 1, indici nelle tabelle di incorporamento.
aggregationWeights Un elenco di tensori di rango 1 contenente i pesi di aggregazione per campione, ovvero per (esempio di addestramento, funzionalità).
modeOverride Una stringa di input che sostituisce la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando è impostato su "unspecified", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzata mode_override.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

Parametri
deviceOrdinal Il dispositivo TPU da utilizzare. Deve essere> = 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo.