clase estática pública EnqueueTPUEmbeddingSparseTensorBatch.Options
Atributos opcionales para EnqueueTPUEmbeddingSparseTensorBatch
Métodos públicos
Poner en colaTPUEmbeddingSparseTensorBatch.Options | combinadores (lista <cadena> combinadores) |
Poner en colaTPUEmbeddingSparseTensorBatch.Options | dispositivoOrdinal (dispositivo largoOrdinal) |
Poner en colaTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (Lista<Long> maxSequenceLengths) |
Métodos heredados
Métodos públicos
Combinadores públicos EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch.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. |
---|