Un'operazione che consente la differenziazione degli incorporamenti TPU.
Questa operazione restituisce semplicemente il suo primo input, che si presuppone sia stato suddiviso dai tensori restituiti da TPUEmbeddingDequeueActivations. La presenza di questa operazione, e il suo primo argomento essendo una variabile addestrabile, consente la differenziazione automatica dei grafici contenenti incorporamenti tramite le librerie Python Embedding TPU.
Metodi pubblici
Uscita <Mobile> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
Attivazioni di incorporamento TPU statiche | |
Uscita <Mobile> | produzione () |
Metodi ereditati
Metodi pubblici
output pubblico <Float> asOutput ()
Restituisce la maniglia simbolica di un tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
Creazione di attivazioni TPUEmbedding statiche pubbliche (ambito ambito , operando <Float> embeddingVariable, operando <Float> slicedActivations, Long tableId, Long lookupId)
Metodo factory per creare una classe che racchiude una nuova operazione TPUEmbeddingActivations.
Parametri
scopo | ambito attuale |
---|---|
incorporamentoVariabile | Una variabile addestrabile, che consente agli ottimizzatori di trovare questa op. |
attivazioni affettate | Il tensore delle attivazioni di incorporamento da restituire. |
tabellaId | L'ID della tabella nella configurazione del livello di incorporamento da cui sono state calcolate queste attivazioni. |
lookupId | Identificatore dell'insieme di indici di incorporamento che hanno prodotto queste attivazioni. |
ritorna
- una nuova istanza di TPUEmbeddingActivations