classe statica pubblica EnqueueTPUEmbeddingBatch.Options
Attributi facoltativi per EnqueueTPUEmbeddingBatch
Metodi pubblici
EnqueueTPUEmbeddingBatch.Options | combinatori (combinatori List<String>) |
EnqueueTPUEmbeddingBatch.Options | dispositivoOrdinale (Dispositivo lungoOrdinale) |
Metodi ereditati
Metodi pubblici
combinatori pubblici EnqueueTPUEmbeddingBatch.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 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. |
---|