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