Facilita la portabilidad 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, 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 colaTPUEmbeddingArbiraryTensorBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingArbitraryTensorBatch | |
Métodos públicos
| EnqueueTPUEmbeddingArbiraryTensorBatch.Options estático | combinadores (lista <cadena> combinadores) |
| estático <T extiende el número, U extiende el número, V extiende el número> EnqueueTPUEmbeddingArbirraryTensorBatch | crear ( alcance alcance , Iterable< Operando <T>> sampleIndicesOrRowSplits, Iterable< Operando <U>> incrustaciónIndices, Iterable< Operando <V>> agregaciónPesos, Operando <Cadena> modeOverride, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingArbiraryTensorBatch. |
| EnqueueTPUEmbeddingArbiraryTensorBatch.Options estático | dispositivoOrdinal (dispositivo largoOrdinal) |
Métodos heredados
Métodos públicos
Combinadores públicos estáticos EnqueueTPUEmbeddingArbiraryTensorBatch.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 EnqueueTPUEmbeddingArbitraryTensorBatch create (alcance del alcance , Iterable < Operando <T>> sampleIndicesOrRowSplits, Iterable < Operando <U>> incrustación de índices, Iterable < Operando <V>> agregaciónPesos, Operando <Cadena> modeOverride, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingArbiraryTensorBatch.
Parámetros
| alcance | alcance actual |
|---|---|
| sampleIndicesOrRowSplits | Una lista de tensores de rango 2 que especifica el ejemplo de entrenamiento al que pertenecen los valores correspondientes de embedding_indices y aggregation_weights. Si el tamaño de su primera dimensión es 0, asumimos que cada embedding_indices 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 filas para dividir embedding_indices y aggregation_weights 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 dividida es un tensor denso 1-D. Cuando está vacío, asumimos que se pasa un tensor denso a la operación. 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. |
| agregaciónPesos | Una lista de tensores de rango 1 que contienen pesos de agregación por ejemplo de entrenamiento. Tanto float32 como float64 están permitidos y se convertirán a float32 internamente. |
| 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. |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de EnqueueTPUEmbeddingArbiraryTensorBatch
público estático EnqueueTPUEmbeddingArbirraryTensorBatch.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. |
|---|