Una operación que recibe activaciones de inserción en la TPU.
El sistema de TPU realiza las búsquedas y agregaciones de incrustación especificadas por los argumentos de TPUEmbeddingEnqueue (Integer / Sparse / SparseTensor) Batch. Los resultados de estas agregaciones son visibles para Tensorflow Graph como los resultados de una operación RecvTPUEmbeddingActivations. Esta operación devuelve una lista que contiene un tensor de activaciones por tabla especificada en el modelo. Puede haber como máximo una operación RecvTPUEmbeddingActivations en el gráfico de TPU.
Métodos públicos
estáticas RecvTPUEmbeddingActivations | |
Iterador < operando <flotador >> | iterador () |
Lista < salida <flotador >> | salidas () Una TensorList de activaciones de inserción que contiene un tensor por tabla de inserción en el modelo. |
Métodos heredados
Métodos públicos
estáticas pública RecvTPUEmbeddingActivations crean ( Alcance alcance, largas numOutputs, Cadena config)
Método de fábrica para crear una clase que envuelva una nueva operación RecvTPUEmbeddingActivations.
Parámetros
alcance | alcance actual |
---|---|
numOutputs | El número de tensores de activación de salida, igual al número de tablas de incrustación en el modelo. |
config | Protocolo TPUEmbeddingConfiguration serializado. |
Devoluciones
- una nueva instancia de RecvTPUEmbeddingActivations
Lista pública < salida <Float >> salidas ()
Una TensorList de activaciones de inserción que contiene un tensor por tabla de inserción en el modelo.