EnqueueTPUEmbeddingArbitraryTensorBatch

clase final pública EnqueueTPUEmbeddingArbitraryTensorBatch

Facilita la transferencia de código que utiliza tf.nn.embedding_lookup_sparse().

embedding_indices[i] y aggregation_weights[i] corresponden a la i-ésima 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 EnqueueTPUEmbeddingArbitraryTensorBatch.Options Atributos opcionales para EnqueueTPUEmbeddingArbitraryTensorBatch

Métodos públicos

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

Métodos Heredados

Métodos públicos

Combinadores públicos estáticos EnqueueTPUEmbeddingArbitraryTensorBatch.Options ( List<String> combinadors)

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 EnqueueTPUEmbeddingArbitraryTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)

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

Parámetros
alcance alcance actual
sampleIndicesOrRowSplits Una lista de tensores de rango 2 que especifican el ejemplo de entrenamiento al que pertenecen los valores de incrustación_índices y agregación_pesos correspondientes. Si el tamaño de su primera dimensión es 0, asumimos que cada incrustación_índices pertenece a una muestra diferente. Tanto int32 como int64 están permitidos y se convertirán a int32 internamente.

O una lista de tensores de rango 1 que especifican las divisiones de fila para dividir incrustaciones_índices y agregaciones_pesos en filas. Corresponde a ids.row_splits en embedding_lookup(), cuando ids es un RaggedTensor. Al poner en cola el tensor irregular ND, solo se permite que la última dimensión sea irregular. la fila se divide es un tensor denso 1-D. Cuando está vacío, asumimos que se pasa un tensor denso a la op Tanto int32 como int64 están permitidos y se convertirán a int32 internamente.

incrustación de índices Una lista de tensores de rango 1, índices en las tablas de incrustación. Tanto int32 como int64 están permitidos y se convertirán a int32 internamente.
Pesos de agregación Una lista de tensores de rango 1 que contienen pesos de agregación de ejemplo de entrenamiento. Tanto float32 como float64 están permitidos y se convertirán a float32 internamente.
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 EnqueueTPUEmbeddingArbitraryTensorBatch

public static EnqueueTPUEmbeddingArbitraryTensorBatch.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.