EnqueueTPUEmbeddingSparseBatch.Options

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.