XlaSparseCoreAdam

パブリック最終クラスXlaSparseCoreAdam

パブリックメソッド

静的XlaSparseCoreAdam
create ( Scopeスコープ、オペランド<Float> embeddingTable、オペランド<Integer> インデックス、オペランド<Float> gradient、オペランド<Float> learningRate、オペランド<Float> 運動量、オペランド<Float> 速度、オペランド<Float> beta1、オペランド<Float > beta2、オペランド<Float> イプシロン、Long featureWidth、ブール型 useSumInsideSqrt)
新しい XlaSparseCoreAdam オペレーションをラップするクラスを作成するファクトリ メソッド。
出力<浮動小数点数>
updatedEmbeddingTable ()
出力<浮動小数点数>
更新された勢い()
出力<浮動小数点数>
更新された速度()

継承されたメソッド

パブリックメソッド

public static XlaSparseCoreAdam create ( Scopeスコープ、オペランド<Float> embeddingTable、オペランド<Integer> インデックス、オペランド<Float> gradient、オペランド<Float> learningRate、オペランド<Float> 運動量、オペランド<Float> 速度、オペランド<Float> beta1、オペランド<Float> beta2、オペランド<Float> epsilon、Long featureWidth、Boolean useSumInsideSqrt)

新しい XlaSparseCoreAdam オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
戻り値
  • XlaSparseCoreAdam の新しいインスタンス

public出力<Float> updatedEmbeddingTable ()

public出力<Float> updatedMomentum ()

public出力<Float> updatedVelocity ()