XlaRecvTPUEmbeddingDeduplicationData

classe finale pubblica XlaRecvTPUEmbeddingDeduplicationData

Riceve i dati di deduplicazione (indici e pesi) dal core di incorporamento.

I dati di deduplicazione sono un tensore con tipo=DT_VARIANT. Il tensore stesso è una tupla annidata XLA contenente N elementi (dove N è il rapporto tra il numero di incorporamenti e i nuclei tensoriali per chip TPU). Ogni elemento della tupla annidata è una tupla di tensori di rango 1. Ogni tensore contiene indici (DT_UINT32) per la ricerca di incorporamento su TensorCore o pesi (DT_FLOAT) da applicare all'output dell'operazione di ricerca di incorporamento.

Metodi pubblici

Uscita <Oggetto>	comeuscita () Restituisce la maniglia simbolica di un tensore.
XlaRecvTPUEmbeddingDeduplicationData statico	creare (ambito ambito , configurazione stringa) Metodo factory per creare una classe che racchiude una nuova operazione XlaRecvTPUEmbeddingDeduplicationData.
Uscita <?>	produzione ()

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 ()

Dall'interfaccia org.tensorflow.Operand

output astratto <Oggetto>

comeuscita () Restituisce la maniglia simbolica di un tensore.

Metodi pubblici

output pubblico <Oggetto> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

creazione XlaRecvTPUEmbeddingDeduplicationData pubblica statica ( ambito ambito , configurazione stringa)

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

Parametri

scopo	ambito attuale
config	Prototipo TPUEmbeddingConfiguration serializzato.

ritorna

• una nuova istanza di XlaRecvTPUEmbeddingDeduplicationData

Uscita pubblica <?> uscita ()