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

EnqueueTPUEmbeddingRaggedTensorBatch

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
EnqueueTPUEmbeddingRaggedTensorBatch public final class

Facilita la migración de código que usa tf.nn.embedding_lookup ().

sample_splits [i], embedding_indices [i] y aggregation_weights [i] corresponden a la i-ésima característica. table_ids [i] indica qué tabla de incrustación buscar en esta función.

Los tensores en las posiciones correspondientes en dos de las listas de entrada, embedding_indices y aggregation_weights, 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 la característica correspondiente.

Clases anidadas

clase EnqueueTPUEmbeddingRaggedTensorBatch.Options Los atributos opcionales para EnqueueTPUEmbeddingRaggedTensorBatch

Métodos públicos

estáticas EnqueueTPUEmbeddingRaggedTensorBatch.Options
combinadores (List <String> combinadores)
estática <T se extiende Número, U se extiende Número, V se extiende Número> EnqueueTPUEmbeddingRaggedTensorBatch
crear ( Alcance alcance, Iterable < operando <T >> sampleSplits, Iterable < operando <U >> embeddingIndices, Iterable < operando <V >> aggregationWeights, operando <String> modeOverride, List <Larga> TableIDs, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingRaggedTensorBatch.
estáticas EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal (Long deviceOrdinal)
estáticas EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (List <Larga> maxSequenceLengths)
estáticas EnqueueTPUEmbeddingRaggedTensorBatch.Options
numFeatures (List <Larga> numFeatures)

Métodos heredados

Métodos públicos

públicas estáticas EnqueueTPUEmbeddingRaggedTensorBatch.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 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.

public static EnqueueTPUEmbeddingRaggedTensorBatch crear ( Alcance alcance, Iterable < operando <T >> sampleSplits, Iterable < Operando embeddingIndices <U >>, Iterable < operando <V >> aggregationWeights, operando <String> modeOverride, List <> TableIDs largas, opciones .. . opciones)

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

Parámetros
alcance alcance actual
sampleSplits Una lista de tensores de rango 1 que especifican los puntos de ruptura para dividir embedding_indices y aggregation_weights en filas. Corresponde a ids.row_splits en embedding_lookup (), cuando ids es un RaggedTensor.
incrustaciónIndices Una lista de tensores de rango 1, índices en las tablas de incrustación. Corresponde a ids.values ​​en embedding_lookup (), cuando ids es un RaggedTensor.
aggregationWeights Una lista de tensores de rango 1 que contiene pesos de agregación por ejemplo de entrenamiento. Corresponde al campo de valores de un RaggedTensor con los mismos row_splits que los ids en embedding_lookup (), cuando ids es un RaggedTensor.
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.
tableIds Una lista de números enteros que especifican el identificador de la tabla de incrustación (desplazamiento de TableDescriptor en TPUEmbeddingConfiguration) para buscar la entrada correspondiente. La i-ésima entrada se busca usando table_ids [i]. El tamaño de la lista table_ids debe ser igual al de sample_indices, embedding_indices y aggregation_weights.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de EnqueueTPUEmbeddingRaggedTensorBatch

public static EnqueueTPUEmbeddingRaggedTensorBatch.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.

estáticas pública EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List <Larga> maxSequenceLengths)

públicas estáticas EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List <Larga> numFeatures)