EmbeddingActivations

EmbeddingActivations della classe finale pubblica

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
create ( Scope scope, Operando < TFloat32 > embeddingVariable, Operando < TFloat32 > slicedActivations, Long tableId, Long lookupId)
Metodo factory per creare una classe che racchiude una nuova operazione EmbeddingActivations.
Uscita < TFloat32 >

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "TPUEmbeddingActivations"

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

Uscita pubblica < TFloat32 > uscita ()