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. |
---|