EnqueueTPUEmbeddingBatch

clase final pública EnqueueTPUEmbeddingBatch

Una operación que pone en cola una lista de tensores por lotes de entrada en TPUEmbedding.

Una operación que pone en cola una lista de tensores por lotes de entrada en TPUEmbedding.

Clases anidadas

clase Poner en colaTPUEmbeddingBatch.Options Atributos opcionales para EnqueueTPUEmbeddingBatch

Métodos públicos

EnqueueTPUEmbeddingBatch.Options estático
combinadores (lista <cadena> combinadores)
En cola estáticaTPUEmbeddingBatch
crear ( alcance alcance , Iterable< Operando <Cadena>> lote, Operando <Cadena> modeOverride, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingBatch.
EnqueueTPUEmbeddingBatch.Options estático
dispositivoOrdinal (dispositivo largoOrdinal)

Métodos heredados

Métodos públicos

Combinadores públicos estáticos EnqueueTPUEmbeddingBatch.Options (combinadores List<String>)

Parámetros
combinadores Una lista de escalares de cadena, uno para cada tabla de incrustación, que especifica cómo normalizar las activaciones de incrustación después de la suma ponderada. Los combinadores admitidos son "mean", "sum" o "sqrtn". No es válido que la suma de los pesos sea 0 para 'media' o que la suma de los pesos al cuadrado sea 0 para 'sqrtn'. Si no se pasan los combinadores, el valor predeterminado es usar 'suma' para todas las tablas.

creación pública estática de EnqueueTPUEmbeddingBatch (alcance del alcance , iterable < operando <cadena>> lote, operando <cadena> modeOverride, opciones... opciones)

Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingBatch.

Parámetros
alcance alcance actual
lote Una lista de tensores 1D, uno para cada tabla de incrustación, que contiene las entradas por lotes codificadas como protos dist_belief.SparseFeatures. Si el campo de peso en el prototipo SparseFeatures no se completa para un ID, se supone un peso de 1,0.
anulación de modo Una entrada de cadena que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'sin especificar', se usa el modo establecido en TPUEmbeddingConfiguration; de lo contrario, se usa mode_override.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de EnqueueTPUEmbeddingBatch

público estático EnqueueTPUEmbeddingBatch.Options dispositivoOrdinal (dispositivo largo Ordinal)

Parámetros
dispositivoOrdinal El dispositivo TPU a utilizar. Esto debe ser -1 cuando el Op se ejecuta en un dispositivo TPU y >= 0 cuando el Op se ejecuta en el dispositivo CPU.