EnqueueTPUEmbeddingSparseBatch.Options

EnqueueTPUEmbeddingSparseBatch.Options estáticos de clase pública

Los atributos opcionales para EnqueueTPUEmbeddingSparseBatch

Métodos públicos

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

Métodos heredados

Métodos públicos

públicas EnqueueTPUEmbeddingSparseBatch.Options combinadores (List <String> combinadores)

Parámetros
combinadores Una lista de escalares de cadena, uno para cada tabla de incrustación que especifica cómo normalizar las activaciones de incrustación después de la suma ponderada. Los combinadores admitidos son 'mean', 'sum' o 'sqrtn'. No es válido que la suma de los pesos sea 0 para 'media' o la suma de los pesos al cuadrado sea 0 para 'sqrtn'. Si no se pasan los combinadores, el valor predeterminado es usar 'suma' para todas las tablas.

pública EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser> = 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo.