EnqueueTPUEmbeddingRaggedTensorBatch

clase final pública EnqueueTPUEmbeddingRaggedTensorBatch

Facilita la portabilidad 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 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 Poner en colaTPUEmbeddingRaggedTensorBatch.Options Atributos opcionales para EnqueueTPUEmbeddingRaggedTensorBatch

Métodos públicos

EnqueueTPUEmbeddingRaggedTensorBatch.Options estático
combinadores (lista <cadena> combinadores)
estático <T extiende el número, U extiende el número, V extiende el número> EnqueueTPUEmbeddingRaggedTensorBatch
crear (alcance de alcance , Iterable< Operando <T>> divisiones de muestra, Iterable< Operando <U>> índices de incrustación, Iterable< Operando <V>> pesos de agregación, Operando <Cadena> modeOverride, Lista<Long> tableIds, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingRaggedTensorBatch.
EnqueueTPUEmbeddingRaggedTensorBatch.Options estático
dispositivoOrdinal (dispositivo largoOrdinal)
EnqueueTPUEmbeddingRaggedTensorBatch.Options estático
maxSequenceLengths (Lista<Long> maxSequenceLengths)
EnqueueTPUEmbeddingRaggedTensorBatch.Options estático
numFeatures (Lista<Larga> numFeatures)

Métodos heredados

Métodos públicos

Combinadores públicos estáticos EnqueueTPUEmbeddingRaggedTensorBatch.Options (combinadores List<String>)

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 create ( Alcance del alcance , Iterable< Operando <T>> sampleSplits, Iterable< Operando <U>> incrustaciónIndices, Iterable< Operando <V>> agregaciónPesos, Operando <Cadena> modeOverride, Lista<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
muestraDivisiones Una lista de tensores de rango 1 que especifican los puntos de interrupción para dividir embedding_indices y aggregation_weights 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.
agregaciónPesos Una lista de tensores de rango 1 que contienen pesos de agregación por ejemplo de entrenamiento. Corresponde al campo de valores de un RaggedTensor con las mismas filas_splits que los identificadores en embedding_lookup(), cuando ids es un RaggedTensor.
anulación de modo Una entrada de cadena que anula 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.
identificadores 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, embedding_indices y aggregation_weights.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de EnqueueTPUEmbeddingRaggedTensorBatch

público estático EnqueueTPUEmbeddingRaggedTensorBatch.Options dispositivoOrdinal (dispositivo largo Ordinal)

Parámetros
dispositivoOrdinal El dispositivo 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.

público estático EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)

público estático EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List<Long> numFeatures)