Facilita a transferência de código que usa tf.nn.embedding_lookup_sparse ().
sample_indices [i], embedding_indices [i] e aggregation_weights [i] correspondem ao iº recurso. table_ids [i] indica qual tabela de incorporação procurar com o recurso.
Os tensores nas posições correspondentes nas três listas de entrada (sample_indices, 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 | EnqueueTPUEmbeddingSparseTensorBatch.Options | Atributos opcionais para EnqueueTPUEmbeddingSparseTensorBatch |
Métodos Públicos
estáticos EnqueueTPUEmbeddingSparseTensorBatch.Options | combinadores (List <String> combinadores) |
estática <T estende Número, U estende Número, V estende Number> EnqueueTPUEmbeddingSparseTensorBatch | criar ( Scope escopo, Iterable < Operando <T >> sampleIndices, 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 EnqueueTPUEmbeddingSparseTensorBatch. |
estáticos EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
estáticos EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (List <longo> maxSequenceLengths) |
estáticos EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (List <longo> numFeatures) |
Métodos herdados
Métodos Públicos
public static EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch criar ( Scope escopo, Iterable < Operando <T >> sampleIndices, 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 EnqueueTPUEmbeddingSparseTensorBatch.
Parâmetros
alcance | escopo atual |
---|---|
sampleIndices | Uma lista de tensores de classificação 1 especificando o exemplo de treinamento ao qual os valores embedding_indices e aggregation_weights correspondentes pertencem. Corresponde a sp_ids.indices [:, 0] em embedding_lookup_sparse (). |
embeddingIndices | Uma lista de tensores de classificação 1, índices nas tabelas de incorporação. Corresponde a sp_ids.values em embedding_lookup_sparse (). |
aggregationWeights | Uma lista de tensores de classificação 1 contendo pesos de agregação por exemplo de treinamento. Corresponde a sp_weights.values em embedding_lookup_sparse (). |
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 EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.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. |
---|