Un'operazione che riceve attivazioni di incorporamento sulla TPU.
Il sistema TPU esegue le ricerche di incorporamento e le aggregazioni specificate dagli argomenti di TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. I risultati di queste aggregazioni sono visibili nel grafico Tensorflow come output di un RecvTPUEmbeddingActivations op. Questa operazione restituisce un elenco contenente un tensore di attivazioni per tabella specificata nel modello. Può essere presente al massimo un'operazione RecvTPUEmbeddingActivations nel grafico TPU.
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Attivazioni RecvTPUEmbedding statiche | |
Iteratore< Operando < TFloat32 >> | iteratore () |
Elenco< Uscita < TFloat32 >> | uscite () Un TensorList di attivazioni di incorporamento contenente un Tensore per tabella di incorporamento nel modello. |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
creazione di RecvTPUEmbeddingActivations statiche pubbliche (ambito ambito , numOutput lunghi, configurazione stringa)
Metodo factory per creare una classe che racchiude una nuova operazione RecvTPUEmbeddingActivations.
Parametri
scopo | ambito attuale |
---|---|
numOutput | Il numero di tensori di attivazione dell'output, pari al numero di tabelle di incorporamento nel modello. |
config | Prototipo TPUEmbeddingConfiguration serializzato. |
ritorna
- una nuova istanza di RecvTPUEmbeddingActivations