EnqueueTPUEmbeddingSparseBatch.Options

EnqueueTPUEmbeddingSparseBatch.Options classe estática pública

Atributos opcionais para EnqueueTPUEmbeddingSparseBatch

Métodos Públicos

EnqueueTPUEmbeddingSparseBatch.Options
combinadores (List <String> combinadores)
EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (Long deviceOrdinal)

Métodos herdados

Métodos Públicos

públicas EnqueueTPUEmbeddingSparseBatch.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.

pública EnqueueTPUEmbeddingSparseBatch.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.