Recibe datos de deduplicación (índices y ponderaciones) del núcleo de incorporación.
Los datos de deduplicación son un tensor con tipo=DT_VARIANT. El tensor en sí es una tupla anidada XLA que contiene N elementos (donde N es la relación entre el número de núcleos de incrustación y tensor 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.
Métodos públicos
| Salida <Objeto> | como salida () Devuelve el identificador simbólico de un tensor. |
| XlaRecvTPUEmbeddingDeduplicationData estático | |
| Salida <?> | producción () |
Métodos heredados
Métodos públicos
Salida pública <Objeto> comoSalida ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
creación pública estática XlaRecvTPUEmbeddingDeduplicationData (alcance alcance , configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingDeduplicationData.
Parámetros
| alcance | alcance actual |
|---|---|
| configuración | Protocolo de configuración de integración de TPU serializado. |
Devoluciones
- una nueva instancia de XlaRecvTPUEmbeddingDeduplicationData