SendTPUEmbeddingGradients

public final class SendTPUEmbeddingGradients

Esegue gli aggiornamenti del gradiente delle tabelle incorporate.

Metodi pubblici

SendTPUEmbeddingGradients statico
create ( ambito ambito, input Iterable < Operand <Float>>, Iterable < Operand <Float>> learningRates, String config)
Metodo di fabbrica per creare una classe che avvolge una nuova operazione SendTPUEmbeddingGradients.

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