EnqueueTPUEmbeddingSparseBatch

classe finale pubblica EnqueueTPUEmbeddingSparseBatch

Un'operazione che accoda TPUEmbedding indici di input da uno SparseTensor.

Questa operazione facilita il porting del codice che utilizza embedding_lookup_sparse(), sebbene sia necessaria una preelaborazione Python degli argomenti SparseTensor su embedding_lookup_sparse() per produrre gli argomenti di questa operazione, poiché è consentita solo una singola operazione EnqueueTPUEmbeddingSparseBatch per fase di training.

I tensori nelle posizioni corrispondenti nelle tre liste di input devono avere la stessa forma, cioè rango 1 con dim_size() uguale al numero totale di ricerche nella tabella descritta dal corrispondente table_id.

Classi nidificate

classe EnqueueTPUEmbeddingSparseBatch.Options Attributi facoltativi per EnqueueTPUEmbeddingSparseBatch

Metodi pubblici

EnqueueTPUEmbeddingSparseBatch.Options statico	combinatori (combinatori List<String>)
static <T estende il numero, U estende il numero, V estende il numero> EnqueueTPUEmbeddingSparseBatch	create ( Scope scope, Iterable< Operando <T>> sampleIndices, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> aggregationWeights, Operando <String> modeOverride, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingSparseBatch.
EnqueueTPUEmbeddingSparseBatch.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

combinatori statici pubblici EnqueueTPUEmbeddingSparseBatch.Options (combinatori List<String>)

Parametri

combinatori	Un elenco di scalari di stringa, uno per ciascuna tabella di incorporamento che specifica come normalizzare le attivazioni di incorporamento dopo la somma ponderata. I combinatori supportati sono "mean", "sum" o "sqrtn". Non è valido che la somma dei pesi sia 0 per "media" o che la somma dei pesi al quadrato sia 0 per "sqrtn". Se non viene passato il combinatore, l'impostazione predefinita è utilizzare 'sum' per tutte le tabelle.

combinatori

Un elenco di scalari di stringa, uno per ciascuna tabella di incorporamento che specifica come normalizzare le attivazioni di incorporamento dopo la somma ponderata. I combinatori supportati sono "mean", "sum" o "sqrtn". Non è valido che la somma dei pesi sia 0 per "media" o che la somma dei pesi al quadrato sia 0 per "sqrtn". Se non viene passato il combinatore, l'impostazione predefinita è utilizzare 'sum' per tutte le tabelle.

public static EnqueueTPUEmbeddingSparseBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)

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

Parametri

scopo	ambito attuale
sampleIndices	Un elenco di tensori di rango 1 che specificano l'esempio di training e la funzionalità a cui appartengono i valori embedding_indices e aggregation_weights corrispondenti. sample_indices[i] deve essere uguale a b * nf + f, dove nf è il numero di funzionalità dalla tabella corrispondente, f è in [0, nf) e b è in [0, dimensione batch).
embeddingIndices	Un elenco di tensori di rango 1, indici nelle tabelle di incorporamento.
aggregationWeights	Un elenco di tensori di rango 1 contenenti pesi di aggregazione per campione, ovvero per (esempio di addestramento, funzionalità).
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 EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEmbeddingSparseBatch.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.