EnqueueTPUEmbeddingIntegerBatch

classe finale pubblica EnqueueTPUEmbeddingIntegerBatch

Un'operazione che accoda un elenco di tensori batch di input a TPUEmbedding.

Classi nidificate

classe

EnqueueTPUEmbeddingIntegerBatch.Options

Attributi facoltativi per EnqueueTPUEmbeddingIntegerBatch

Metodi pubblici

EnqueueTPUEmbeddingIntegerBatch statico	create ( Scope scope, Iterable< Operando <Integer>> batch, Operando <String> modeOverride, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingIntegerBatch.
EnqueueTPUEmbeddingIntegerBatch.Options statico	dispositivoOrdinale (Dispositivo lungoOrdinale)

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Oggetto oggetto)
finale int	codice hash ()
Operazione	operazione () Restituisce l' Operation sottostante
stringa finale	accordare ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
Classe finale<?>	getClass ()
int	codice hash ()
vuoto finale	notificare ()
vuoto finale	notificaTutti ()
Corda	accordare ()
vuoto finale	attendere (lungo arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	Aspettare ()

Metodi pubblici

public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options)

Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingIntegerBatch.

Parametri

scopo	ambito attuale
lotto	Un elenco di tensori 1D, uno per ciascuna tabella di incorporamento, contenente gli indici nelle tabelle.
modeOverride	Un input di tipo stringa che sovrascrive la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Se impostato su "non specificato", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzato mode_override.
opzioni	trasporta valori di attributi opzionali

ritorna

• una nuova istanza di EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Parametri

dispositivoOrdinale	Il dispositivo TPU da utilizzare. Dovrebbe essere >= 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo.