clase estática pública EnqueueTPUEmbeddingSparseBatch.Options
Atributos opcionales para EnqueueTPUEmbeddingSparseBatch
Métodos públicos
| Poner en colaTPUEmbeddingSparseBatch.Options | combinadores (lista <cadena> combinadores) |
| Poner en colaTPUEmbeddingSparseBatch.Options | dispositivoOrdinal (dispositivo largoOrdinal) |
Métodos heredados
Métodos públicos
Combinadores públicos EnqueueTPUEmbeddingSparseBatch.Options (combinadores List<String>)
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 que 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. |
|---|
public EnqueueTPUEmbeddingSparseBatch.Options dispositivoOrdinal (dispositivo largo Ordinal)
Parámetros
| dispositivoOrdinal | El dispositivo 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. |
|---|