パブリック最終クラスResourceSparseApplyKerasMomentum
運動量スキームに従って、「* var」および「* accum」の関連エントリを更新します。
Nesterovの勢いを使用する場合は、use_nesterov = Trueに設定します。
これは、卒業した行の場合、varとaccumを次のように更新します。
accum = accum *運動量-lr * grad var + = accum
ネストされたクラス
クラス | ResourceSparseApplyKerasMomentum.Options | ResourceSparseApplyKerasMomentum オプションの属性 |
パブリックメソッド
static <T、U extends Number> ResourceSparseApplyKerasMomentum | |
静的ResourceSparseApplyKerasMomentum.Options | useLocking (ブール値useLocking) |
静的ResourceSparseApplyKerasMomentum.Options | useNesterov (ブール値useNesterov) |
継承されたメソッド
パブリックメソッド
public static ResourceSparseApplyKerasMomentum create (スコープスコープ、オペランド<?> var、オペランド<?> accum、オペランド<T> lr、オペランド<T> grad、オペランド<U>インデックス、オペランド<T>モメンタム、オプション...オプション)
新しいResourceSparseApplyKerasMomentum操作をラップするクラスを作成するファクトリメソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
var | Variable()からのものである必要があります。 |
アキュム | Variable()からのものである必要があります。 |
lr | 学習率。スカラーである必要があります。 |
卒業生 | グラデーション。 |
インデックス | varとaccumの最初の次元へのインデックスのベクトル。 |
勢い | 勢い。スカラーである必要があります。 |
オプション | オプションの属性値を運ぶ |
戻り値
- ResourceSparseApplyKerasMomentumの新しいインスタンス
public static ResourceSparseApplyKerasMomentum.Options useLocking (ブール値useLocking)
パラメーター
useLocking | `True`の場合、varテンソルとaccumテンソルの更新はロックによって保護されます。それ以外の場合、動作は定義されていませんが、競合が少なくなる可能性があります。 |
---|
public static ResourceSparseApplyKerasMomentum.Options useNesterov (ブール値useNesterov)
パラメーター
useNesterov | `True`の場合、gradを計算するために渡されるテンソルはvar + momentum * accumになるため、最終的に、取得するvarは実際にはvar + momentum * accumになります。 |
---|