SendTPUEmbeddingGradients

パブリック最終クラスSendTPUEmbeddingGradients

埋め込みテーブルの勾配更新を実行します。

パブリックメソッド

静的SendTPUEmbeddingGradients
create (スコープscope、Iterable< Operand <Float>> inputs、Iterable< Operand <Float>> learningRates、String config)
新しい 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 の新しいインスタンス