clase final pública XlaRecvTPUEmbeddingActivations
Una operación que recibe activaciones de incrustación en la TPU.
El sistema de TPU realiza las búsquedas y agregaciones incrustadas. Los resultados de estas agregaciones son visibles para 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 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
Public static XlaRecvTPUEmbeddingActivations create ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingActivations.
Parámetros
| alcance | alcance actual |
|---|---|
| datos de deduplicación | Un tensor con type=DT_VARIANT que contiene los datos de deduplicación. El tensor es una tupla anidada XLA que contiene N elementos (donde N es la proporción del número de incrustaciones a 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 incrustación de búsqueda. |
| numTablas | 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 | Proto de configuración de integración de TPU serializado. |
Devoluciones
- una nueva instancia de XlaRecvTPUEmbeddingActivations
public List< Output <Float>> salidas ()
Una TensorList de activaciones de incrustación que contiene un Tensor por tabla de incrustación en el modelo.