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
| booleano finale | è uguale a (Oggetto oggetto) |
| finale int | codice hash () |
| Operazione | operazione () Restituisce questa unità di calcolo come una singola Operation . |
| stringa finale | accordare () |
| 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 () |
| ambiente di esecuzione astratto | ambiente () Restituisce l'ambiente di esecuzione in cui è stata creata questa operazione. |
| Operazione astratta | operazione () Restituisce questa unità di calcolo come una singola Operation . |
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. |
|---|