Um op que enfileira índices de entrada TPUEmbedding de um SparseTensor.
Este Op facilita a portabilidade do código que usa embedding_lookup_sparse(), embora seja necessário algum pré-processamento em Python dos argumentos SparseTensor para embedding_lookup_sparse() para produzir os argumentos para este Op, já que apenas um único EnqueueTPUEmbeddingSparseBatch Op é permitido por etapa de treinamento.
Os tensores nas posições correspondentes nas três listas de entrada devem ter a mesma forma, ou seja, rank 1 com dim_size() igual ao número total de pesquisas na tabela descrita pelo table_id correspondente.
Classes aninhadas
classe | EnqueueTPUEembeddingSparseBatch.Options | Atributos opcionais para EnqueueTPUEmbeddingSparseBatch |
Métodos públicos
static EnqueueTPUEembeddingSparseBatch.Options | combinadores (List<String> combinadores) |
estático <T estende o número, U estende o número, V estende o número> EnqueueTPUEmbeddingSparseBatch | create ( Escopo do escopo, Iterable< Operando <T>> sampleIndices, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> aggregationWeights, Operando <String> modeOverride, Options... options) Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingSparseBatch. |
static EnqueueTPUEembeddingSparseBatch.Options | deviceOrdinal (Longo deviceOrdinal) |
Métodos Herdados
Métodos públicos
public static EnqueueTPUEmbeddingSparseBatch.Options combinadores (List<String> combinadores)
Parâmetros
combinadores | Uma lista de escalares de string, um para cada tabela de incorporação que especifica como normalizar as ativações de incorporação após a soma ponderada. Os combinadores suportados são 'mean', 'sum' ou 'sqrtn'. É inválido que a soma dos pesos seja 0 para 'média' ou a soma dos pesos quadrados seja 0 para 'sqrtn'. Se os combinadores não forem passados, o padrão é usar 'sum' para todas as tabelas. |
---|
public static EnqueueTPUEmbeddingSparseBatch create ( Escopo do escopo, Iterable< Operando <T>> sampleIndices, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> aggregationWeights, Operando <String> modeOverride, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingSparseBatch.
Parâmetros
alcance | escopo atual |
---|---|
índices de amostra | Uma lista de tensores de classificação 1 especificando o exemplo de treinamento e o recurso ao qual os valores embedding_indices e aggregation_weights correspondentes pertencem. sample_indices[i] deve ser igual a b * nf + f, onde nf é o número de recursos da tabela correspondente, f está em [0, nf) e b está em [0, tamanho do lote). |
embeddingIndices | Uma lista de tensores de nível 1, índices nas tabelas de incorporação. |
pesos de agregação | Uma lista de Tensores de classificação 1 contendo por amostra -- ou seja, por (exemplo de treinamento, recurso) -- pesos de agregação. |
modeOverride | Uma entrada de string que substitui o modo especificado no TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando definido como 'unspecified', o modo definido em TPUEmbeddingConfiguration é usado, caso contrário, mode_override é usado. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de EnqueueTPUEmbeddingSparseBatch
public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)
Parâmetros
dispositivoOrdinal | O dispositivo TPU a ser usado. Deve ser >= 0 e menor que o número de núcleos de TPU na tarefa na qual o nó é colocado. |
---|