パブリック最終クラスSendTPUEmbeddingGradients
埋め込みテーブルの勾配更新を実行します。
定数
| 弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
| 静的SendTPUEmbeddingGradients |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "SendTPUEmbeddingGradients"
パブリックメソッド
public static SendTPUEmbeddingGradients create (スコープスコープ、Iterable< Operand < TFloat32 >> 入力、Iterable< Operand < TFloat32 >> learningRates、String config)
新しい SendTPUEmbeddingGradients オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 入力 | 埋め込みテーブルを更新するための勾配の TensorList。この引数は、RecvTPUEmbeddingActivations の戻り値と同じ長さと形状を持ちますが、埋め込みアクティベーションに関するモデルの損失の勾配が含まれています。埋め込みテーブルは、tpu.initialize_system に指定された TPU 埋め込み構成で指定されたオプティマイザを介して、これらの勾配から更新されます。 |
| 学習率 | float32 スカラーの TensorList (動的学習率タグごとに 1 つ): //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto のコメントを参照してください。複数のテーブルは、構成で指定されている同じ動的学習率タグを共有できます。すべてのテーブルの学習率が一定である場合、このリストは空である必要があります。 |
| 構成 | シリアル化された TPUEmbeddingConfiguration プロト。 |
戻り値
- SendTPUEmbeddingGradients の新しいインスタンス