Facilita la migración de código que usa tf.nn.embedding_lookup_sparse ().
sample_indices [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 las tres listas de entrada (sample_indices, 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 | EnqueueTPUEmbeddingSparseTensorBatch.Options | Los atributos opcionales para EnqueueTPUEmbeddingSparseTensorBatch |
Métodos públicos
estáticas EnqueueTPUEmbeddingSparseTensorBatch.Options | combinadores (List <String> combinadores) |
estática <T se extiende Número, U se extiende Número, V se extiende Número> EnqueueTPUEmbeddingSparseTensorBatch | crear ( Alcance alcance, Iterable < operando <T >> sampleIndices, 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 EnqueueTPUEmbeddingSparseTensorBatch. |
estáticas EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
estáticas EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (List <Larga> maxSequenceLengths) |
estáticas EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (List <Larga> numFeatures) |
Métodos heredados
Métodos públicos
públicas estáticas EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch crear ( Alcance alcance, Iterable < operando <T >> sampleIndices, Iterable < operando <T >> embeddingIndices, Iterable < operando <V >> aggregationWeights, operando <String> modeOverride, Lista <pulsación larga> TableIDs, Opciones .. . opciones)
Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingSparseTensorBatch.
Parámetros
alcance | alcance actual |
---|---|
sampleIndices | Una lista de tensores de rango 1 que especifica el ejemplo de entrenamiento al que pertenecen los valores de embedding_indices y aggregation_weights correspondientes. Corresponde a sp_ids.indices [:, 0] en embedding_lookup_sparse (). |
incrustaciónIndices | Una lista de tensores de rango 1, índices en las tablas de incrustación. Corresponde a sp_ids.values en embedding_lookup_sparse (). |
aggregationWeights | Una lista de tensores de rango 1 que contiene pesos de agregación por ejemplo de entrenamiento. Corresponde a sp_weights.values en embedding_lookup_sparse (). |
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 EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.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. |
---|