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.

Metodi pubblici

Uscita <Mobile>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
Attivazioni di incorporamento TPU statiche
create ( Ambito ambito , Operando <Float> embeddingVariable, Operando <Float> slicedActivations, Long tableId, Long lookupId)
Metodo factory per creare una classe che racchiude una nuova operazione TPUEmbeddingActivations.
Uscita <Mobile>

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

Uscita pubblica <Float> uscita ()