SendTPUEmbeddingGradients

パブリック最終クラスSendTPUEmbeddingGradients

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

定数

弦

OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

静的SendTPUEmbeddingGradients

create (スコープスコープ、Iterable< Operand < TFloat32 >> 入力、Iterable< Operand < TFloat32 >> learningRates、String config) 新しい SendTPUEmbeddingGradients オペレーションをラップするクラスを作成するファクトリ メソッド。

継承されたメソッド

クラスorg.tensorflow.op.RawOpから

最終ブール値	等しい(オブジェクトオブジェクト)
最終整数	ハッシュコード（）
手術	オプ() この計算単位を単一のOperationとして返します。
最後の文字列	toString ()

クラスjava.lang.Objectから

ブール値	等しい(オブジェクト arg0)
最終クラス<?>	getクラス()
整数	ハッシュコード（）
最後の空白	通知する()
最後の空白	すべて通知()
弦	toString ()
最後の空白	wait (long arg0, int arg1)
最後の空白	待機(長い引数0)
最後の空白	待って（）

インターフェースorg.tensorflow.op.Opから

抽象的な実行環境	環境() このオペレーションが作成された実行環境を返します。
抽象的な操作	オプ() この計算単位を単一のOperationとして返します。

定数

パブリック静的最終文字列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 の新しいインスタンス