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
RecvTPUE estáticombeddingActivations | crear ( alcance de alcance, numOutputs largos, configuración de cadena) Método de fábrica para crear una clase que envuelva una nueva operación RecvTPUEmbeddingActivations. |
Iterador < Operando <Float>> | iterador () |
Lista < Salida <Float>> | 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
RecvTPUEmbeddingActivations estáticos públicos crear ( alcance de alcance, numOutputs largos, configuración de cadena)
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
Public List < Output <Float>> salidas ()
Una TensorList de activaciones de inserción que contiene un tensor por tabla de inserción en el modelo.