clase estática pública EnqueueTPUEmbeddingSparseBatch.Options
Atributos opcionales para EnqueueTPUEmbeddingSparseBatch
Métodos públicos
EnqueueTPUEmbeddingSparseBatch.Options | combinadores (Lista<String> combinadores) |
EnqueueTPUEmbeddingSparseBatch.Options | deviceOrdinal (dispositivo largo ordinal) |
Métodos Heredados
Métodos públicos
Combinadores públicos EnqueueTPUEmbbeddingSparseBatch.Options ( combinadores List<String>)
Parámetros
combinadores | Una lista de cadenas escalares, una 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 que la suma de los pesos cuadrados sea 0 para 'sqrtn'. Si no se pasan los combinadores, el valor predeterminado es usar 'suma' para todas las tablas. |
---|
public EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)
Parámetros
dispositivoordinal | El dispositivo de TPU a utilizar. Debe ser >= 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo. |
---|