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. |
---|