EnqueueTPUEmbeddingBatch.Options

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.