EnqueueTPUEmbeddingIntegerBatch

public final class EnqueueTPUEmbeddingIntegerBatch

Un op che accoda un elenco di tensori batch di input per TPUEmbedding.

Classi annidate

classe

EnqueueTPUEmbeddingIntegerBatch.Options

Attributi facoltativi per EnqueueTPUEmbeddingIntegerBatch

Metodi pubblici

static EnqueueTPUEmbeddingIntegerBatch	create ( ambito ambito, Iterable < Operando <Integer>> batch, Operando <String> modeOverride, Opzioni ... opzioni) Metodo Factory per creare una classe che avvolge una nuova operazione EnqueueTPUEmbeddingIntegerBatch.
static EnqueueTPUEmbeddingIntegerBatch.Options	deviceOrdinal (dispositivo lungoOrdinal)

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Object obj)
final int	hashCode ()
Operazione	op () Restituisce l' Operation sottostante
stringa finale	toString ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
classe finale <?>	getClass ()
int	hashCode ()
vuoto finale	notifica ()
vuoto finale	notifyAll ()
Corda	toString ()
vuoto finale	wait (long arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	aspetta ()

Metodi pubblici

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

Metodo Factory per creare una classe che avvolge 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	Una stringa di input che sostituisce la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando è impostato su "unspecified", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzata mode_override.
opzioni	trasporta valori di attributi opzionali

ritorna

• una nuova istanza di EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Parametri

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