Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

EnqueueTPUEmbeddingRaggedTensorBatch

EnqueueTPUEmbeddingRaggedTensorBatch classe final pública

Facilita a transferência de código que usa tf.nn.embedding_lookup ().

sample_splits [i], embedding_indices [i] e aggregation_weights [i] correspondem à i-ésima característica. 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 aggregation_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

classe EnqueueTPUEmbeddingRaggedTensorBatch.Options Atributos opcionais para EnqueueTPUEmbeddingRaggedTensorBatch

Métodos Públicos

estáticos EnqueueTPUEmbeddingRaggedTensorBatch.Options
combinadores (List <String> combinadores)
estática <T estende Número, U estende Número, V estende Number> EnqueueTPUEmbeddingRaggedTensorBatch
criar ( Scope escopo, Iterable < Operando <T >> sampleSplits, Iterable < Operando <u >> embeddingIndices, Iterable < Operando <V >> aggregationWeights, Operando <String> modeOverride, List <longo> TableIDs, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingRaggedTensorBatch.
estáticos EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal (Long deviceOrdinal)
estáticos EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (List <longo> maxSequenceLengths)
estáticos EnqueueTPUEmbeddingRaggedTensorBatch.Options
numFeatures (List <longo> numFeatures)

Métodos herdados

Métodos Públicos

public static EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch criar ( Scope escopo, Iterable < Operando <T >> sampleSplits, Iterable < Operando <U >> embeddingIndices, Iterable < Operando <V >> aggregationWeights, Operando <String> modeOverride, List <longo> TableIDs, Opções .. . opções)

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

Parâmetros
alcance escopo atual
sampleSplits Uma lista de tensores de classificação 1 especificando os pontos de quebra para dividir embedding_indices e aggregation_weights em linhas. Corresponde a ids.row_splits em embedding_lookup (), quando ids é um RaggedTensor.
embeddingIndices 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.
aggregationWeights 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 o mesmo row_splits como ids em embedding_lookup (), quando ids é um RaggedTensor.
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.
tableIds Uma lista de inteiros especificando o identificador da tabela de incorporação (deslocamento de TableDescriptor em TPUEmbeddingConfiguration) para pesquisar a entrada correspondente. A enésima entrada é pesquisada usando table_ids [i]. O tamanho da lista table_ids deve ser igual ao de sample_indices, embedding_indices e aggregation_weights.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de EnqueueTPUEmbeddingRaggedTensorBatch

public static EnqueueTPUEmbeddingRaggedTensorBatch.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.

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

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List <longo> numFeatures)