TPUEmbeddingActivations

classe finale pubblica TPUEmbeddingActivations

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 TPU 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 TPUEmbeddingActivations.
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 TPUEmbeddingActivations statiche pubbliche (ambito ambito , operando < TFloat32 > embeddingVariable, operando < TFloat32 > 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

Uscita pubblica < TFloat32 > uscita ()