Conozca lo último en aprendizaje automático, IA generativa y más en el Simposio WiML 2023.

Se usó la API de Cloud Translation para traducir esta página.

EnqueueTPUEmbeddingIntegerBatch

EnqueueTPUEmbeddingIntegerBatch public final class

Una operación que pone en cola una lista de tensores de lote de entrada en TPUEmbedding.

clase EnqueueTPUEmbeddingIntegerBatch.Options Los atributos opcionales para EnqueueTPUEmbeddingIntegerBatch

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

estática EnqueueTPUEmbeddingIntegerBatch	crear ( Alcance alcance, Iterable < operando < TInt32 >> lotes, operando < TString > modeOverride, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingIntegerBatch.
estáticas EnqueueTPUEmbeddingIntegerBatch.Options	deviceOrdinal (Long deviceOrdinal)

booleano final	es igual a (Object obj)
int final	hashCode ()
Operación	op () Devuelva esta unidad de cálculo como una sola `Operation` .
cadena final	toString ()

De la clase java.lang.Object

abstracta ExecutionEnvironment	env () Devuelve el entorno de ejecución en el que se creó esta operación.
abstracta Operación	op () Devuelva esta unidad de cálculo como una sola `Operation` .

Constantes

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "EnqueueTPUEmbeddingIntegerBatch"

Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingIntegerBatch.

alcance	alcance actual
lote	Una lista de tensores 1D, uno para cada tabla de incrustación, que contiene los índices en las tablas.
modeOverride	Una cadena de entrada que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'no especificado', se usa el modo establecido en TPUEmbeddingConfiguration; de lo contrario, se usa mode_override.
opciones	lleva valores de atributos opcionales

deviceOrdinal	El dispositivo de TPU que se utilizará. Debe ser> = 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo.