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.
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita < TFloat32 > | comeuscita () Restituisce l'handle simbolico del tensore. |
Attivazioni di incorporamento statiche | |
Uscita < TFloat32 > | produzione () |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica < TFloat32 > asOutput ()
Restituisce l'handle simbolico del 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 di incorporamento statiche pubbliche (ambito di applicazione , operando < TFloat32 > embeddingVariable, operando < TFloat32 > slicedActivations, ID tabella lungo, ID ricerca lunga)
Metodo factory per creare una classe che racchiude una nuova operazione EmbeddingActivations.
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 EmbeddingActivations