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 | |
Costanti
| Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
| EnqueueTPUEmbeddingIntegerBatch statico | |
| EnqueueTPUEmbeddingIntegerBatch.Options statico | dispositivoOrdinale (Dispositivo lungoOrdinale) |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Valore costante: "EnqueueTPUEmbeddingIntegerBatch"
Metodi pubblici
public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > 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. |
|---|