EnqueueTPUEmbeddingRaggedTensorBatch

classe finale pubblica EnqueueTPUEmbeddingRaggedTensorBatch

Facilita il porting del codice che utilizza tf.nn.embedding_lookup().

sample_splits[i], embedding_indices[i] e aggregation_weights[i] corrispondono alla i-esima caratteristica. table_ids[i] indica quale tabella di incorporamento cercare con la funzione.

I tensori nelle posizioni corrispondenti in due delle liste di input, embedding_indices e aggregation_weights, devono avere la stessa forma, cioè rango 1 con dim_size() uguale al numero totale di ricerche nella tabella descritta dalla funzione corrispondente.

Classi nidificate

classe EnqueueTPUEmbeddingRaggedTensorBatch.Options Attributi facoltativi per EnqueueTPUEmbeddingRaggedTensorBatch

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

EnqueueTPUEmbeddingRaggedTensorBatch.Options statico
combinatori (combinatori List<String>)
static <T estende TNumber , U estende TNumber , V estende TNumber > EnqueueTPUEmbeddingRaggedTensorBatch
create ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options... options)
Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingRaggedTensorBatch.
EnqueueTPUEmbeddingRaggedTensorBatch.Options statico
dispositivoOrdinale (Dispositivo lungoOrdinale)
EnqueueTPUEmbeddingRaggedTensorBatch.Options statico
maxSequenceLengths (Elenco<lungo> maxSequenceLengths)

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "EnqueueTPUEmbeddingRaggedTensorBatch"

Metodi pubblici

combinatori pubblici statici EnqueueTPUEmbeddingRaggedTensorBatch.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.

public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. . opzioni)

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

Parametri
scopo ambito attuale
campioneSplits Un elenco di tensori di rango 1 che specificano i punti di interruzione per dividere embedding_indices e aggregation_weights in righe. Corrisponde a ids.row_splits in embedding_lookup(), quando ids è un RaggedTensor.
embeddingIndices Un elenco di tensori di rango 1, indici nelle tabelle di incorporamento. Corrisponde a ids.values ​​in embedding_lookup(), quando ids è un RaggedTensor.
aggregationWeights Un elenco di tensori di rango 1 contenenti pesi di aggregazione per esempio di training. Corrisponde al campo valori di un RaggedTensor con gli stessi row_splits di ids in embedding_lookup(), quando ids è un RaggedTensor.
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.
tableId Un elenco di numeri interi che specificano l'identificatore della tabella di incorporamento (offset di TableDescriptor in TPUEmbeddingConfiguration) per cercare l'input corrispondente. L'i-esimo input viene cercato utilizzando table_ids[i]. La dimensione dell'elenco table_ids deve essere uguale a quella di sample_indices, embedding_indices e aggregation_weights.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di EnqueueTPUEmbeddingRaggedTensorBatch

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

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)