EnqueueTPUEmbeddingSparseBatch

classe final pública EnqueueTPUEmbeddingSparseBatch

Uma operação que enfileira índices de entrada TPUEmbedding de um SparseTensor.

Esta operação 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 operação, uma vez que apenas uma única operação EnqueueTPUEmbeddingSparseBatch é permitida 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

aula EnqueueTPUEmbeddingSparseBatch.Options Atributos opcionais para EnqueueTPUEmbeddingSparseBatch

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

estático EnqueueTPUEmbeddingSparseBatch.Options
combinadores (combinadores List<String>)
static <T estende TNumber , U estende TNumber , V estende TNumber > EnqueueTPUEmbeddingSparseBatch
create ( Escopo de escopo , Iterable< Operando <T>> sampleIndices, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> gregationWeights, Operando < TString > modeOverride, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingSparseBatch.
estático EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (dispositivo longo Ordinal)

Métodos herdados

org.tensorflow.op.RawOp
booleano final
é igual (objeto obj)
int final
código hash ()
Operação
op ()
Retorne esta unidade de cálculo como uma única Operation .
sequência final
para sequenciar ()
boleano
é igual (objeto arg0)
aula final<?>
getClass ()
interno
código hash ()
vazio final
notificar ()
vazio final
notificar todos ()
Corda
para sequenciar ()
vazio final
espere (long arg0, int arg1)
vazio final
espere (arg0 longo)
vazio final
espere ()
org.tensorflow.op.Op
ambiente de execução abstrato
env ()
Retorne o ambiente de execução em que esta operação foi criada.
operação abstrata
op ()
Retorne esta unidade de cálculo como uma única Operation .

Constantes

String final estática pública OP_NAME

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

Valor constante: "EnqueueTPUEmbeddingSparseBatch"

Métodos Públicos

combinadores public static EnqueueTPUEmbeddingSparseBatch.Options (combinadores List<String>)

Parâmetros
combinadores Uma lista de escalares de cadeia de caracteres, 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 que a soma dos pesos quadrados seja 0 para 'sqrtn'. Se os combinadores não forem passados, o padrão é usar 'soma' para todas as tabelas.

public static EnqueueTPUEmbeddingSparseBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> gregationWeights, Operand < TString > modeOverride, Options... options)

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

Parâmetros
escopo escopo atual
índices de amostra Uma lista de tensores de classificação 1 especificando o exemplo de treinamento e o recurso ao qual pertencem os valores embedding_indices e agregation_weights correspondentes. 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).
incorporação de índices Uma lista de tensores de classificação 1, índices nas tabelas de incorporação.
agregaçãoPesos Uma lista de tensores de classificação 1 contendo por amostra - ou seja, por (exemplo de treinamento, recurso) - pesos de agregação.
modoSubstituir Uma entrada de string que substitui o modo especificado em TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', '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
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ó está colocado.