clase estática pública EnqueueTPUEmbeddingBatch.Options
Atributos opcionales para EnqueueTPUEmbeddingBatch
Métodos públicos
Poner en colaTPUEmbeddingBatch.Options | combinadores (lista <cadena> combinadores) |
Poner en colaTPUEmbeddingBatch.Options | dispositivoOrdinal (dispositivo largoOrdinal) |
Métodos heredados
Métodos públicos
Combinadores públicos EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch.Options dispositivoOrdinal (dispositivo largo)
Parámetros
dispositivoOrdinal | El dispositivo TPU a utilizar. Esto debe ser -1 cuando el Op se ejecuta en un dispositivo TPU y >= 0 cuando el Op se ejecuta en el dispositivo CPU. |
---|