classe statica pubblica EnqueueTPUEmbeddingSparseTensorBatch.Options
Attributi facoltativi per EnqueueTPUEmbeddingSparseTensorBatch
Metodi pubblici
EnqueueTPUEmbeddingSparseTensorBatch.Options | combinatori (List <String> combiners) |
EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (dispositivo lungoOrdinal) |
EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (List <Long> maxSequenceLengths) |
Metodi ereditati
Metodi pubblici
public EnqueueTPUEmbeddingSparseTensorBatch.Options combiners (List <String> combiners)
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 la somma dei pesi al quadrato sia 0 per "sqrtn". Se i combinatori non vengono passati, l'impostazione predefinita è utilizzare "sum" per tutte le tabelle. |
---|
public EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Parametri
deviceOrdinal | Il dispositivo TPU da utilizzare. Deve essere> = 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo. |
---|