XlaRecvTPUEmbeddingActivations

classe finale pubblica XlaRecvTPUEmbeddingActivations

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
create ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)
Metodo factory per creare una classe che racchiude una nuova operazione XlaRecvTPUEmbeddingActivations.
Iteratore< Operando <Float>>
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

iteratore pubblico< operando <float>> iteratore ()

public List< Output <Float>> uscite ()

Un TensorList di attivazioni di incorporamento contenente un Tensore per tabella di incorporamento nel modello.