EnqueueTPUEmbeddingBatch

clase final pública EnqueueTPUEmbeddingBatch

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

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

Clases anidadas

clase EnqueueTPUEmbeddingBatch.Options Atributos opcionales para EnqueueTPUEmbeddingBatch

Métodos públicos

static EnqueueTPUEmbeddingBatch.Options
combinadores (Lista<String> combinadores)
static EnqueueTPUEmbeddingBatch
crear ( Ámbito de ámbito, 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.
static EnqueueTPUEmbeddingBatch.Options
deviceOrdinal (dispositivo largo ordinal)

Métodos Heredados

Métodos públicos

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

Parámetros
combinadores Una lista de cadenas escalares, una 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 cuadrados sea 0 para 'sqrtn'. Si no se pasan los combinadores, el valor predeterminado es usar 'suma' para todas las tablas.

public static EnqueueTPUEmbeddingBatch create ( Scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)

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 ponderación en el prototipo de SparseFeatures no se completa para una ID, se asume una ponderación de 1.0.
modeOverride Una entrada de cadena que invalida 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

public static EnqueueTPUEmbeddingBatch.Options dispositivo ordinal ( dispositivo largo ordinal)

Parámetros
dispositivoordinal El dispositivo de TPU a utilizar. Debe ser -1 cuando la operación se ejecuta en un dispositivo de TPU y >= 0 cuando la operación se ejecuta en el dispositivo de CPU.