Un'operazione che riceve le attivazioni di incorporamento sulla TPU.
Il sistema TPU esegue le ricerche di incorporamento e le aggregazioni specificate dagli argomenti per 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ò esserci al massimo un'operazione RecvTPUEmbeddingActivations nel grafico TPU.
Metodi pubblici
static RecvTPUEmbeddingActivations | |
Iteratore < Operando <Float>> | iteratore () |
Elenco < Output <Float>> | uscite () Un TensorList di attivazioni di incorporamento contenente un Tensor per tabella di incorporamento nel modello. |
Metodi ereditati
Metodi pubblici
public static RecvTPUEmbeddingActivations create ( Scope scope, Long numOutputs, String config)
Metodo Factory per creare una classe che avvolge una nuova operazione RecvTPUEmbeddingActivations.
Parametri
scopo | ambito attuale |
---|---|
numOutputs | Il numero di tensori di attivazione dell'uscita, pari al numero di tabelle di incorporamento nel modello. |
config | Protocollo TPUEmbeddingConfiguration serializzato. |
ritorna
- una nuova istanza di RecvTPUEmbeddingActivations
public List < Output <Float>> output ()
Un TensorList di attivazioni di incorporamento contenente un Tensor per tabella di incorporamento nel modello.