Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

EnqueueTPUEmbeddingSparseBatch

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
clase final pública EnqueueTPUEmbeddingSparseBatch

Una operación que pone en cola los índices de entrada de TPUEmbedding de un SparseTensor.

Esta operación facilita la portabilidad del código que usa embedding_lookup_sparse(), aunque se requiere algo de preprocesamiento de Python de los argumentos de SparseTensor para embedding_lookup_sparse() para producir los argumentos de esta operación, ya que solo se permite una sola operación EnqueueTPUEmbeddingSparseBatch por paso de entrenamiento.

Los tensores en las posiciones correspondientes en las tres listas de entrada 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 el table_id correspondiente.

Clases anidadas

clase EnqueueTPUEmbeddingSparseBatch.Options Atributos opcionales para EnqueueTPUEmbeddingSparseBatch

Métodos públicos

static EnqueueTPUEmbeddingSparseBatch.Options
combinadores (Lista<String> combinadores)
estático <T extiende el número, U extiende el número, V extiende el número> EnqueueTPUEmbeddingSparseBatch
create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> incrustarIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingSparseBatch.
static EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (dispositivo largo ordinal)

Métodos Heredados

Métodos públicos

combinadores públicos estáticos EnqueueTPUEmbeddingSparseBatch.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 EnqueueTPUEmbeddingSparseBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> incrustingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)

Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingSparseBatch.

Parámetros
alcance alcance actual
índices de muestra Una lista de tensores de rango 1 que especifican el ejemplo de entrenamiento y la función a la que pertenecen los valores de incrustación_índices y agregación_pesos correspondientes. sample_indices[i] debe ser igual a b * nf + f, donde nf es el número de características de la tabla correspondiente, f está en [0, nf) y b está en [0, tamaño del lote).
incrustación de índices Una lista de tensores de rango 1, índices en las tablas de incrustación.
Pesos de agregación Una lista de tensores de rango 1 que contienen por muestra, es decir, por (ejemplo de entrenamiento, función), pesos de agregación.
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.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

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.