clase final pública EnqueueTPUEmbeddingRaggedTensorBatch
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. |
|---|