EnqueueTPUEmbeddingRaggedTensorBatch

classe final pública EnqueueTPUEmbeddingRaggedTensorBatch

Facilita a portabilidade de código que usa tf.nn.embedding_lookup().

sample_splits[i], embedding_indices[i] e gregation_weights[i] correspondem ao i-ésimo recurso. table_ids[i] indica qual tabela de incorporação procurar com o recurso.

Os tensores nas posições correspondentes em duas das listas de entrada, embedding_indices e agregation_weights, devem ter a mesma forma, ou seja, classificação 1 com dim_size() igual ao número total de pesquisas na tabela descrita pelo recurso correspondente.

Classes aninhadas

aula EnqueueTPUEmbeddingRaggedTensorBatch.Options Atributos opcionais para EnqueueTPUEmbeddingRaggedTensorBatch

Constantes

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

Métodos Públicos

estático EnqueueTPUEmbeddingRaggedTensorBatch.Options
combinadores (combinadores List<String>)
estático <T estende TNumber , U estende TNumber , V estende TNumber > EnqueueTPUEmbeddingRaggedTensorBatch
create ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> gregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options... options)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingRaggedTensorBatch.
estático EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal (dispositivo longo Ordinal)
estático EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (Lista<Long> maxSequenceLengths)

Métodos herdados

Constantes

String final estática pública OP_NAME

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

Valor constante: "EnqueueTPUEmbeddingRaggedTensorBatch"

Métodos Públicos

combinadores public static EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> gregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. .opções )

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

Parâmetros
escopo escopo atual
amostraSplits Uma lista de tensores de classificação 1 especificando os pontos de interrupção para dividir embedding_indices e agregation_weights em linhas. Corresponde a ids.row_splits em embedding_lookup(), quando ids é um RaggedTensor.
incorporação de índices Uma lista de tensores de classificação 1, índices nas tabelas de incorporação. Corresponde a ids.values ​​em embedding_lookup(), quando ids é um RaggedTensor.
agregaçãoPesos Uma lista de tensores de classificação 1 contendo pesos de agregação por exemplo de treinamento. Corresponde ao campo de valores de um RaggedTensor com os mesmos row_splits que ids em embedding_lookup(), quando ids é um RaggedTensor.
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.
IDs da tabela Uma lista de inteiros especificando o identificador da tabela de incorporação (deslocamento de TableDescriptor no TPUEmbeddingConfiguration) para consultar a entrada correspondente. A i-ésima entrada é consultada usando table_ids[i]. O tamanho da lista table_ids deve ser igual ao de sample_indices, embedding_indices e gregation_weights.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de EnqueueTPUEmbeddingRaggedTensorBatch

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (dispositivo longoOrdinal)

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.

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)