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