Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

XlaRecvTPUEmbeddingActivations

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
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
crear ( Ámbito de alcance, Operando <?> deduplicationData, Long numTables, String config)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingActivations.
Iterador< Operando <Flotante>>
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 Iterator< Operando <Flotante>> iterador ()

public List< Output <Float>> salidas ()

Una TensorList de activaciones de incrustación que contiene un Tensor por tabla de incrustación en el modelo.