classe estática pública EnqueueTPUEmbeddingSparseBatch.Options
Atributos opcionais para EnqueueTPUEmbeddingSparseBatch
Métodos públicos
EnqueueTPUEembeddingSparseBatch.Options | combinadores (List<String> combinadores) |
EnqueueTPUEembeddingSparseBatch.Options | deviceOrdinal (Longo deviceOrdinal) |
Métodos Herdados
Métodos públicos
public 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 os combinadores não forem passados, o padrão é usar 'sum' para todas as tabelas. |
---|
public EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)
Parâmetros
dispositivoOrdinal | O dispositivo TPU a ser usado. Deve ser >= 0 e menor que o número de núcleos de TPU na tarefa na qual o nó é colocado. |
---|