EnqueueTPUEmbeddingSparseBatch

EnqueueTPUEmbeddingSparseBatch classe final pública

Um op que enfileira índices de entrada TPUEmbedding de um SparseTensor.

Esta Op facilita a portabilidade de código que usa embedding_lookup_sparse (), embora algum pré-processamento Python dos argumentos SparseTensor para embedding_lookup_sparse () seja necessário para produzir os argumentos para esta Op, uma vez 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, classificação 1 com dim_size () igual ao número total de pesquisas na tabela descrita pelo table_id correspondente.

Classes aninhadas

classe EnqueueTPUEmbeddingSparseBatch.Options Atributos opcionais para EnqueueTPUEmbeddingSparseBatch

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

estáticos EnqueueTPUEmbeddingSparseBatch.Options
combinadores (List <String> combinadores)
estática <T estende TNumber , U estende TNumber , V estende TNumber > EnqueueTPUEmbeddingSparseBatch
criar ( Scope escopo, Iterable < Operando <T >> sampleIndices, Iterable < Operando <u >> embeddingIndices, Iterable < Operando <V >> aggregationWeights, Operando < TString > modeOverride, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingSparseBatch.
estáticos EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (Long deviceOrdinal)

Métodos herdados

Constantes

nome_op final String public static

O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Valor constante: "EnqueueTPUEmbeddingSparseBatch"

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 combinadores não forem passados, o padrão é usar 'soma' para todas as tabelas.

public static EnqueueTPUEmbeddingSparseBatch criar ( Scope escopo, Iterable < Operando <T >> sampleIndices, Iterable < Operando <u >> embeddingIndices, Iterable < Operando <V >> aggregationWeights, Operando < TString > modeOverride, Options ... Opções)

Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingSparseBatch.

Parâmetros
alcance escopo atual
sampleIndices Uma lista de tensores de classificação 1 especificando o exemplo de treinamento e o recurso aos quais 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 classificação 1, índices nas tabelas de incorporação.
aggregationWeights 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 na TPUEmbeddingConfiguration. Os valores suportados são {'não especificado', 'inferência', 'treinamento', 'backward_pass_only'}. Quando definido como 'não especificado', 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
deviceOrdinal O dispositivo TPU a ser usado. Deve ser> = 0 e menor que o número de núcleos de TPU na tarefa em que o nó está colocado.