EnqueueTPUEmbeddingSparseBatch

EnqueueTPUEmbeddingSparseBatch public final class

Una operación que pone en cola TPUEincorporando índices de entrada de un SparseTensor.

Esta operación facilita la portabilidad del código que usa embedding_lookup_sparse (), aunque se requiere cierto procesamiento previo de Python de los argumentos de SparseTensor a embedding_lookup_sparse () para producir los argumentos de esta operación, ya que solo se permite una sola operación EnqueueTPUEmbeddingSparseBatch por paso de entrenamiento.

Los tensores en las posiciones correspondientes en las tres listas de entrada deben tener la misma forma, es decir, rango 1 con dim_size () igual al número total de búsquedas en la tabla descrita por el table_id correspondiente.

Clases anidadas

clase EnqueueTPUEmbeddingSparseBatch.Options Los atributos opcionales para EnqueueTPUEmbeddingSparseBatch

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estáticas EnqueueTPUEmbeddingSparseBatch.Options
combinadores (List <String> combinadores)
estática <T se extiende TNumber , U se extiende TNumber , V se extiende TNumber > EnqueueTPUEmbeddingSparseBatch
crear ( Alcance alcance, Iterable < operando <T >> sampleIndices, Iterable < operando <U >> embeddingIndices, Iterable < operando <V >> aggregationWeights, operando < TString > modeOverride, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingSparseBatch.
estáticas EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (Long deviceOrdinal)

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "EnqueueTPUEmbeddingSparseBatch"

Métodos públicos

públicas estáticas EnqueueTPUEmbeddingSparseBatch.Options combinadores (List <String> combinadores)

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 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.

public static EnqueueTPUEmbeddingSparseBatch crear ( Alcance alcance, Iterable < operando <T >> sampleIndices, Iterable < operando <U >> embeddingIndices, Iterable < operando <V >> aggregationWeights, operando < TString > modeOverride, Opciones ... Opciones)

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

Parámetros
alcance alcance actual
sampleIndices Una lista de tensores de rango 1 que especifican el ejemplo de entrenamiento y la característica a la que pertenecen los valores de embedding_indices y aggregation_weights correspondientes. sample_indices [i] debe ser igual a b * nf + f, donde nf es el número de características de la tabla correspondiente, f está en [0, nf) y b está en [0, tamaño de lote).
incrustaciónIndices Una lista de tensores de rango 1, índices en las tablas de incrustación.
aggregationWeights Una lista de tensores de rango 1 que contiene por muestra, es decir, por (ejemplo de entrenamiento, función), ponderaciones de agregación.
modeOverride Una cadena de entrada que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'no especificado', 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 EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser> = 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo.