Un'operazione che riceve attivazioni di incorporamento sulla TPU.
Il sistema TPU esegue le ricerche e le aggregazioni di incorporamento. I risultati di queste aggregazioni sono visibili nel grafico Tensorflow come output di un XlaRecvTPUEmbeddingActivations Op. Questa operazione restituisce un elenco contenente un tensore di attivazioni per tabella specificata nel modello.
Metodi pubblici
Attivazioni statiche XlaRecvTPUEmbedding | |
Iteratore< Operando <Float>> | iteratore () |
Elenco< Uscita <Float>> | uscite () Un TensorList di attivazioni di incorporamento contenente un Tensore per tabella di incorporamento nel modello. |
Metodi ereditati
Metodi pubblici
creazione XlaRecvTPUEmbeddingActivations pubblica statica ( ambito ambito, operando <?> deduplicationData, tabelle num lunghe, configurazione stringa)
Metodo factory per creare una classe che racchiude una nuova operazione XlaRecvTPUEmbeddingActivations.
Parametri
scopo | ambito attuale |
---|---|
deduplicazioneData | Un tensore con tipo=DT_VARIANT contenente i dati di deduplicazione. Il tensore è una tupla annidata XLA contenente N elementi (dove N è il rapporto tra il numero di incorporamenti e i nuclei tensoriali per chip TPU). Ogni elemento della tupla annidata è una tupla di tensori di rango 1. Ogni tensore contiene indici (DT_UINT32) per la ricerca di incorporamento su TensorCore o pesi (DT_FLOAT) da applicare all'output dell'operazione di ricerca di incorporamento. |
numTabelle | Il numero di tensori di attivazione dell'uscita. Se il descrittore della funzionalità è presente nella configurazione di incorporamento della TPU, è uguale al numero di funzionalità altrimenti pari al numero di tabelle di incorporamento nel modello. |
config | Prototipo TPUEmbeddingConfiguration serializzato. |
ritorna
- una nuova istanza di XlaRecvTPUEmbeddingActivations
public List< Output <Float>> uscite ()
Un TensorList di attivazioni di incorporamento contenente un Tensore per tabella di incorporamento nel modello.