RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

public Final class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

デバッグサポートを使用して Momentum 埋め込みパラメータを取得します。

ホストメモリへの埋め込みから最適化パラメータを取得する演算。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作はチェックポイントを保存する前に更新されたパラメータを取得するために使用されます。

ネストされたクラス

クラスRetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options RetrieveTPUEmbeddingMomentumParametersGradAccumDebugのオプション属性

パブリックメソッド

static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
config (文字列構成)
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
create (スコープscope、Long numShards、Long shardId、オプション...オプション)
新しい RetrieveTPUEmbeddingMomentumParametersGradAccumDebug 操作をラップするクラスを作成するファクトリ メソッド。
出力<浮動小数点>
勾配アキュムレータ()
パラメーター gradient_accumulators は、Momentum 最適化アルゴリズムによって更新されます。
出力<浮動小数点>
勢い()
運動量最適化アルゴリズムによって更新されるパラメーター運動量。
出力<浮動小数点>
パラメーター()
パラメータ パラメータは、Momentum 最適化アルゴリズムによって更新されます。
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (長いテーブル ID)
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
テーブル名(文字列テーブル名)

継承されたメソッド

パブリックメソッド

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options config (文字列構成)

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug create (スコープscope、Long numShards、Long shardId、 Options...オプション)

新しい RetrieveTPUEmbeddingMomentumParametersGradAccumDebug 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
オプションオプションの属性値を持ちます
戻り値
  • RetrieveTPUEmbeddingMomentumParametersGradAccumDebug の新しいインスタンス

public Output <Float> gradientAccumulators ()

パラメーター gradient_accumulators は、Momentum 最適化アルゴリズムによって更新されます。

public出力<Float> momenta ()

運動量最適化アルゴリズムによって更新されるパラメータ運動量。

public出力<Float>パラメータ()

パラメータ パラメータは、Momentum 最適化アルゴリズムによって更新されます。

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)