clase final pública XlaRecvTPUEmbeddingActivations
Una operación que recibe activaciones integradas en la TPU.
El sistema TPU realiza búsquedas y agregaciones de incorporación. Los resultados de estas agregaciones son visibles en Tensorflow Graph como las salidas de una operación XlaRecvTPUEmbeddingActivations. Esta operación devuelve una lista que contiene un tensor de activaciones por tabla especificada en el modelo.
Métodos públicos
| Activaciones estáticas de XlaRecvTPUEmbedding | |
| Iterador< Operando <Flotante>> | iterador () |
| Lista< Salida <Flotante>> | salidas () Una TensorList de activaciones de incrustación que contiene un tensor por tabla de incrustación en el modelo. |
Métodos heredados
Métodos públicos
creación pública estática de XlaRecvTPUEmbeddingActivations (alcance de alcance , operando <?> deduplicationData, Long numTables, configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingActivations.
Parámetros
| alcance | alcance actual |
|---|---|
| deduplicaciónDatos | Un tensor con tipo=DT_VARIANT que contiene los datos de deduplicación. El tensor es una tupla anidada XLA que contiene N elementos (donde N es la relación entre el número de incrustaciones y núcleos tensoriales por chip de TPU). Cada elemento de la tupla anidada es una tupla de tensores de rango 1. Cada tensor contiene índices (DT_UINT32) para incrustar la búsqueda en TensorCore o pesos (DT_FLOAT) para aplicar a la salida de la operación de búsqueda de incrustación. |
| numTables | El número de tensores de activación de salida. Si el descriptor de características está presente en la configuración de incrustación de tpu, es igual a la cantidad de características; de lo contrario, es igual a la cantidad 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 XlaRecvTPUEmbeddingActivations
Lista pública< Salida <Float>> salidas ()
Una TensorList de activaciones de incrustación que contiene un tensor por tabla de incrustación en el modelo.