Un'operazione che accoda TPUEmbedding indici di input da uno SparseTensor.
Questa operazione facilita il porting del codice che utilizza embedding_lookup_sparse(), sebbene sia necessaria una preelaborazione Python degli argomenti SparseTensor su embedding_lookup_sparse() per produrre gli argomenti di questa operazione, poiché è consentita solo una singola operazione EnqueueTPUEmbeddingSparseBatch per fase di training.
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 nidificate
classe | EnqueueTPUEmbeddingSparseBatch.Options | Attributi facoltativi per EnqueueTPUEmbeddingSparseBatch |
Metodi pubblici
EnqueueTPUEmbeddingSparseBatch.Options statico | combinatori (combinatori List<String>) |
static <T estende il numero, U estende il numero, V estende il numero> EnqueueTPUEmbeddingSparseBatch | create ( Scope scope, Iterable< Operando <T>> sampleIndices, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> aggregationWeights, Operando <String> modeOverride, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingSparseBatch. |
EnqueueTPUEmbeddingSparseBatch.Options statico | dispositivoOrdinale (Dispositivo lungoOrdinale) |
Metodi ereditati
Metodi pubblici
combinatori statici pubblici EnqueueTPUEmbeddingSparseBatch.Options (combinatori List<String>)
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 che la somma dei pesi al quadrato sia 0 per "sqrtn". Se non viene passato il combinatore, 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, Options... options)
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 specificano l'esempio di training e la funzionalità 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 funzionalità dalla tabella corrispondente, f è in [0, nf) e b è in [0, dimensione batch). |
embeddingIndices | Un elenco di tensori di rango 1, indici nelle tabelle di incorporamento. |
aggregationWeights | Un elenco di tensori di rango 1 contenenti pesi di aggregazione per campione, ovvero per (esempio di addestramento, funzionalità). |
modeOverride | Un input di tipo stringa che sovrascrive la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Se impostato su "non specificato", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzato mode_override. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di EnqueueTPUEmbeddingSparseBatch
public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)
Parametri
dispositivoOrdinale | Il dispositivo TPU da utilizzare. Dovrebbe essere >= 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo. |
---|