RetrieveTPUEmbeddingADAMParameters

パブリック最終クラスRetrieveTPUEmbeddingADAMParameters

ADAM 埋め込みパラメータを取得します。

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

ネストされたクラス

クラスTPUEmbeddingADAMParameters.Options の取得RetrieveTPUEmbeddingADAMParametersのオプションの属性

定数

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

パブリックメソッド

静的RetrieveTPUEmbeddingADAMParameters.Options
config (文字列構成)
静的RetrieveTPUEmbeddingADAMParameters
create (スコープscope、Long numShards、Long shardId、オプション...オプション)
新しい RetrieveTPUEmbeddingADAMParameters 操作をラップするクラスを作成するためのファクトリ メソッド。
出力< TFloat32 >
勢い()
ADAM 最適化アルゴリズムによって更新されるパラメーターの運動量。
出力< TFloat32 >
パラメーター()
パラメータ ADAM 最適化アルゴリズムによって更新されるパラメータ。
静的RetrieveTPUEmbeddingADAMParameters.Options
tableId (長いテーブル ID)
静的RetrieveTPUEmbeddingADAMParameters.Options
テーブル名(文字列テーブル名)
出力< TFloat32 >
速度()
ADAM 最適化アルゴリズムによって更新されたパラメーター速度。

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

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

定数値: "RetrieveTPUEmbeddingADAMParameters"

パブリックメソッド

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

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

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

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

public出力< TFloat32 > momenta ()

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

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

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

public static RetrieveTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParameters.Options tableName (String tableName)

public出力< TFloat32 >速度()

ADAM 最適化アルゴリズムによって更新されたパラメーター速度。