Facilita la transferencia de código que utiliza 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 con la característica.
Los tensores en las posiciones correspondientes en las tres listas de entrada (sample_indices, incrustar_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 función correspondiente.
Clases anidadas
clase | EnqueueTPUEmbeddingSparseTensorBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingSparseTensorBatch |
Métodos públicos
static EnqueueTPUEmbeddingSparseTensorBatch.Options | combinadores (Lista<String> combinadores) |
estático <T extiende el número, U extiende el número, V extiende el número> EnqueueTPUEmbeddingSparseTensorBatch | create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> incrustación de índices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options... options) Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingSparseTensorBatch. |
static EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (dispositivo largo ordinal) |
static EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (Lista <Long> maxSequenceLengths) |
static EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (Lista<Long> numFeatures) |
Métodos Heredados
Métodos públicos
combinadores públicos estáticos EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> incrustarIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options.. . opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingSparseTensorBatch.
Parámetros
alcance | alcance actual |
---|---|
índices de muestra | Una lista de tensores de rango 1 que especifican el ejemplo de entrenamiento al que pertenecen los valores de incrustación_índices y agregación_pesos correspondientes. Corresponde a sp_ids.indices[:,0] en embedding_lookup_sparse(). |
incrustación de índices | Una lista de tensores de rango 1, índices en las tablas de incrustación. Corresponde a sp_ids.values en embedding_lookup_sparse(). |
Pesos de agregación | Una lista de tensores de rango 1 que contienen pesos de agregación de ejemplo de entrenamiento. Corresponde a sp_weights.values en embedding_lookup_sparse(). |
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. |
ID de tabla | 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, incrustar_índices y aggregation_weights. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.Options dispositivo ordinal ( dispositivo largo ordinal)
Parámetros
dispositivoordinal | El dispositivo de TPU a utilizar. Debe ser >= 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo. |
---|