Una operación que recibe activaciones integradas en la TPU.
El sistema TPU realiza las búsquedas de incrustación y las agregaciones especificadas por los argumentos de TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. Los resultados de estas agregaciones son visibles para Tensorflow Graph como 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.
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
RecvTPUEmbeddingActivaciones estáticas | |
Iterador< Operando < TFloat32 >> | iterador () |
Lista < Salida < TFloat32 >> | salidas () Una TensorList de activaciones de incrustación que contiene un tensor por tabla de incrustación en el modelo. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Creación de RecvTPUEmbeddingActivations estáticas públicas (alcance de alcance , salidas numéricas largas, configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación RecvTPUEmbeddingActivations.
Parámetros
alcance | alcance actual |
---|---|
numSalidas | El número de tensores de activación de salida, igual al número de tablas de incrustación en el modelo. |
configuración | Protocolo de configuración de integración de TPU serializado. |
Devoluciones
- una nueva instancia de RecvTPUEmbeddingActivations