Una operación que pone en cola una lista de tensores por lotes de entrada para TPUEmbedding.
Una operación que pone en cola una lista de tensores por lotes de entrada para TPUEmbedding.
Clases anidadas
clase | EnqueueTPUEmbeddingBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingBatch |
Métodos públicos
static EnqueueTPUEmbeddingBatch.Options | combinadores (Lista<String> combinadores) |
static EnqueueTPUEmbeddingBatch | crear ( Ámbito de ámbito, 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. |
static EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (dispositivo largo ordinal) |
Métodos Heredados
Métodos públicos
Combinadores públicos estáticos EnqueueTPUEmbeddingBatch.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 static EnqueueTPUEmbeddingBatch create ( Scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)
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 ponderación en el prototipo de SparseFeatures no se completa para una ID, se asume una ponderación de 1.0. |
modeOverride | Una entrada de cadena que invalida 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
public static EnqueueTPUEmbeddingBatch.Options dispositivo ordinal ( dispositivo largo ordinal)
Parámetros
dispositivoordinal | El dispositivo de TPU a utilizar. Debe ser -1 cuando la operación se ejecuta en un dispositivo de TPU y >= 0 cuando la operación se ejecuta en el dispositivo de CPU. |
---|