public final class SendTPUEmbeddingGradients
Esegue gli aggiornamenti del gradiente delle tabelle incorporate.
Metodi pubblici
SendTPUEmbeddingGradients statico |
Metodi ereditati
Metodi pubblici
public static SendTPUEmbeddingGradients create ( Scope scope, Iterable < Operand <Float>> input, Iterable < Operand <Float>> learningRates, String config)
Metodo di fabbrica per creare una classe che avvolge una nuova operazione SendTPUEmbeddingGradients.
Parametri
scopo | ambito attuale |
---|---|
ingressi | Un TensorList di gradienti con cui aggiornare le tabelle di incorporamento. Questo argomento ha la stessa lunghezza e forme del valore restituito di RecvTPUEmbeddingActivations, ma contiene gradienti di perdita del modello rispetto alle attivazioni di incorporamento. Le tabelle di incorporamento vengono aggiornate da questi gradienti tramite l'ottimizzatore specificato nella configurazione di incorporamento TPU data a tpu.initialize_system. |
learningRates | Un TensorList di scalari float32, uno per ogni tag del tasso di apprendimento dinamico: vedere i commenti in //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Più tabelle possono condividere lo stesso tag del tasso di apprendimento dinamico specificato nella configurazione. Se i tassi di apprendimento per tutte le tabelle sono costanti, l'elenco dovrebbe essere vuoto. |
config | Protocollo TPUEmbeddingConfiguration serializzato. |
ritorna
- una nuova istanza di SendTPUEmbeddingGradients