EnqueueTPUEmbeddingBatch.Options

classe estática pública EnqueueTPUEmbeddingBatch.Options

Atributos opcionais para EnqueueTPUEmbeddingBatch

Métodos públicos

EnqueueTPUEmbeddingBatch.Options
combinadores (List<String> combinadores)
EnqueueTPUEmbeddingBatch.Options
deviceOrdinal (Longo deviceOrdinal)

Métodos Herdados

Métodos públicos

Combinadores públicos EnqueueTPUEmbeddingBatch.Options (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 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.