EnqueueTPUEmbeddingBatch

classe final pública EnqueueTPUEmbeddingBatch

Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.

Um op 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

static EnqueueTPUEmbeddingBatch.Options
combinadores (List<String> combinadores)
static EnqueueTPUEmbeddingBatch
create ( Escopo do escopo, Iterable< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)
Método de fábrica para criar uma classe envolvendo uma nova operação EnqueueTPUEmbeddingBatch.
static EnqueueTPUEmbeddingBatch.Options
deviceOrdinal (Longo deviceOrdinal)

Métodos Herdados

Métodos públicos

public static EnqueueTPUEmbeddingBatch.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 os combinadores não forem passados, o padrão é usar 'sum' para todas as tabelas.

public static EnqueueTPUEmbeddingBatch create ( Escopo do escopo, Iterable< Operando <String>> lote, Operando <String> modeOverride, Opções... opções)

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

Parâmetros
alcance escopo atual
lote Uma lista de tensores 1D, um para cada tabela de incorporação, contendo as entradas de lote codificadas como protos dist_belief.SparseFeatures. Se o campo de peso no proto SparseFeatures não for preenchido para um ID, um peso de 1,0 será assumido.
modeOverride Uma entrada de string que substitui o modo especificado no TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando definido como 'unspecified', 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 está sendo executado em um dispositivo TPU e >= 0 quando o Op está sendo executado no dispositivo CPU.