EnqueueTPUEmbeddingBatch

classe final pública EnqueueTPUEmbeddingBatch

Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.

Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.

Classes aninhadas

aula EnqueueTPUEmbeddingBatch.Options Atributos opcionais para EnqueueTPUEmbeddingBatch

Métodos Públicos

EnqueueTPUEmbeddingBatch.Options estático
combinadores (combinadores List<String>)
EnqueueTPUEmbeddingBatch estático
create ( Escopo , Iterável< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingBatch.
EnqueueTPUEmbeddingBatch.Options estático
deviceOrdinal (dispositivo longo Ordinal)

Métodos herdados

Métodos Públicos

combinadores public static EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch create ( Escopo de escopo , Iterável< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)

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

Parâmetros
escopo escopo atual
lote Uma lista de tensores 1D, um para cada tabela de incorporação, contendo as entradas em lote codificadas como protos dist_belief.SparseFeatures. Se o campo de peso no proto SparseFeatures não estiver preenchido para um ID, um peso de 1,0 será assumido.
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 EnqueueTPUEmbeddingBatch

public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)

Parâmetros
dispositivoOrdinal O dispositivo TPU a ser usado. Deve ser -1 quando o Op estiver sendo executado em um dispositivo TPU e >= 0 quando o Op estiver sendo executado no dispositivo CPU.