Una operación que pone en cola una lista de tensores por lotes de entrada en TPUEmbedding.
Una operación que pone en cola una lista de tensores por lotes de entrada en TPUEmbedding.
Clases anidadas
clase | Poner en colaTPUEmbeddingBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingBatch |
Métodos públicos
EnqueueTPUEmbeddingBatch.Options estático | combinadores (lista <cadena> combinadores) |
En cola estáticaTPUEmbeddingBatch | crear ( alcance alcance , Iterable< Operando <Cadena>> lote, Operando <Cadena> modeOverride, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingBatch. |
EnqueueTPUEmbeddingBatch.Options estático | dispositivoOrdinal (dispositivo largoOrdinal) |
Métodos heredados
Métodos públicos
Combinadores públicos estáticos 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. |
---|
creación pública estática de EnqueueTPUEmbeddingBatch (alcance del alcance , iterable < operando <cadena>> lote, operando <cadena> modeOverride, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingBatch.
Parámetros
alcance | alcance actual |
---|---|
lote | Una lista de tensores 1D, uno para cada tabla de incrustación, que contiene las entradas por lotes codificadas como protos dist_belief.SparseFeatures. Si el campo de peso en el prototipo SparseFeatures no se completa para un ID, se supone un peso de 1,0. |
anulación de modo | Una entrada de cadena que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'sin especificar', se usa el modo establecido en TPUEmbeddingConfiguration; de lo contrario, se usa mode_override. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de EnqueueTPUEmbeddingBatch
público estático EnqueueTPUEmbeddingBatch.Options dispositivoOrdinal (dispositivo largo Ordinal)
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. |
---|