SendTPUEmbeddingGradients
Mantieni tutto organizzato con le raccolte
Salva e classifica i contenuti in base alle tue preferenze.
Esegue aggiornamenti gradienti delle tabelle di incorporamento.
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi ereditati
Dalla classe java.lang.Object booleano | è uguale a (Oggetto arg0) |
Classe finale<?> | getClass () |
int | codice hash () |
vuoto finale | notificare () |
vuoto finale | notificaTutti () |
Corda | accordare () |
vuoto finale | attendere (lungo arg0, int arg1) |
vuoto finale | aspetta (lungo arg0) |
vuoto finale | Aspettare () |
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Valore costante: "SendTPUEmbeddingGradients"
Metodi pubblici
Metodo factory per creare una classe che racchiude una nuova operazione SendTPUEmbeddingGradients.
Parametri
scopo | ambito attuale |
---|
input | Un TensorList di gradienti con cui aggiornare le tabelle di incorporamento. Questo argomento ha la stessa lunghezza e forma 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 fornita a tpu.initialize_system. |
---|
Tassi di apprendimento | Un TensorList di scalari float32, uno per ogni tag di tasso di apprendimento dinamico: vedere i commenti in //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Più tabelle possono condividere lo stesso tag di tasso di apprendimento dinamico come specificato nella configurazione. Se i tassi di apprendimento per tutte le tabelle sono costanti, questo elenco dovrebbe essere vuoto. |
---|
config | Prototipo TPUEmbeddingConfiguration serializzato. |
---|
ritorna
- una nuova istanza di SendTPUEmbeddingGradients
Salvo quando diversamente specificato, i contenuti di questa pagina sono concessi in base alla licenza Creative Commons Attribution 4.0, mentre gli esempi di codice sono concessi in base alla licenza Apache 2.0. Per ulteriori dettagli, consulta le norme del sito di Google Developers. Java è un marchio registrato di Oracle e/o delle sue consociate.
Ultimo aggiornamento 2025-07-26 UTC.
[null,null,["Ultimo aggiornamento 2025-07-26 UTC."],[],[],null,["# SendTPUEmbeddingGradients\n\npublic final class **SendTPUEmbeddingGradients** \nPerforms gradient updates of embedding tables. \n\n### Constants\n\n|--------|---------------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|--------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients) | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#create(org.tensorflow.op.Scope, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.String))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config) Factory method to create a class wrapping a new SendTPUEmbeddingGradients operation. |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"SendTPUEmbeddingGradients\"\n\nPublic Methods\n--------------\n\n#### public static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients)\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config)\n\nFactory method to create a class wrapping a new SendTPUEmbeddingGradients operation. \n\n##### Parameters\n\n| scope | current scope |\n| inputs | A TensorList of gradients with which to update embedding tables. This argument has the same length and shapes as the return value of RecvTPUEmbeddingActivations, but contains gradients of the model's loss with respect to the embedding activations. The embedding tables are updated from these gradients via the optimizer specified in the TPU embedding configuration given to tpu.initialize_system. |\n| learningRates | A TensorList of float32 scalars, one for each dynamic learning rate tag: see the comments in //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Multiple tables can share the same dynamic learning rate tag as specified in the configuration. If the learning rates for all tables are constant, this list should be empty. |\n| config | Serialized TPUEmbeddingConfiguration proto. |\n|---------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of SendTPUEmbeddingGradients"]]