Un'operazione che accoda un elenco di tensori batch di input a TPUEmbedding.
Un'operazione che accoda un elenco di tensori batch di input a TPUEmbedding.
Classi nidificate
classe | EnqueueTPUEmbeddingBatch.Options | Attributi facoltativi per EnqueueTPUEmbeddingBatch |
Metodi pubblici
EnqueueTPUEmbeddingBatch.Options statico | combinatori (combinatori List<String>) |
EnqueueTPUEmbeddingBatch statico | create ( ambito ambito , Iterable< operando <String>> batch, operando <String> modeOverride, opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingBatch. |
EnqueueTPUEmbeddingBatch.Options statico | dispositivoOrdinale (Dispositivo lungoOrdinale) |
Metodi ereditati
Metodi pubblici
combinatori EnqueueTPUEmbeddingBatch.Options statici pubblici (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 EnqueueTPUEmbeddingBatch create ( scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)
Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingBatch.
Parametri
scopo | ambito attuale |
---|---|
lotto | Un elenco di tensori 1D, uno per ogni tabella di incorporamento, contenente gli input batch codificati come prototipi dist_belief.SparseFeatures. Se il campo peso nel protocollo SparseFeatures non è popolato per un ID, si presuppone un peso pari a 1,0. |
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 EnqueueTPUEmbeddingBatch
public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
Parametri
dispositivoOrdinale | Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando l'Operazione è in esecuzione su un dispositivo TPU e >= 0 quando l'Operazione è in esecuzione sul dispositivo CPU. |
---|