EnqueueTPUEmbeddingSparseBatch.Options

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.