Facilita la transferencia de código que utiliza 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 con la característica.
Los tensores en las posiciones correspondientes en dos de las listas de entrada, incrustar_índices y agregación_pesos, 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 | Atributos opcionales para EnqueueTPUEmbeddingRaggedTensorBatch |
Métodos públicos
static EnqueueTPUEmbeddingRaggedTensorBatch.Options | combinadores (Lista<String> combinadores) |
estático <T extiende el número, U extiende el número, V extiende el número> EnqueueTPUEmbeddingRaggedTensorBatch | create ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, 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 EnqueueTPUEmbeddingRaggedTensorBatch. |
static EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (dispositivo largo ordinal) |
static EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (Lista <Long> maxSequenceLengths) |
static EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (Lista<Long> numFeatures) |
Métodos Heredados
Métodos públicos
combinadores públicos estáticos EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch create ( Ámbito de ámbito, Iterable< Operando <T>> divisiones de muestra, Iterable< Operando <U>> incrustar Índices, Iterable< Operando <V>> Pesos de agregación, Operando <String> modeOverride, List<Long> tableIds, Opciones.. . opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingRaggedTensorBatch.
Parámetros
alcance | alcance actual |
---|---|
divisiones de muestra | Una lista de tensores de rango 1 que especifican los puntos de ruptura para dividir incrustaciones_índices y agregaciones_pesos en filas. Corresponde a ids.row_splits en embedding_lookup(), cuando ids es un RaggedTensor. |
incrustación de índices | 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. |
Pesos de agregación | Una lista de tensores de rango 1 que contienen pesos de agregación de ejemplo de entrenamiento. Corresponde al campo de valores de un RaggedTensor con las mismas divisiones de filas que los ids en embedding_lookup(), cuando ids es un RaggedTensor. |
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 EnqueueTPUEmbeddingRaggedTensorBatch
public static EnqueueTPUEmbeddingRaggedTensorBatch.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. |
---|