パブリック最終クラス
運動量スキームに従って「*var」を更新します。
Nesterov の勢いを使用したい場合は、use_nesterov = True を設定します。
accum = accum * 勢い - lr * grad var += accum
ネストされたクラス
クラス | リソースApplyKerasMomentum.Options | ResourceApplyKerasMomentum のオプションの属性 |
パブリックメソッド
静的 <T> ResourceApplyKerasMomentum | |
静的ResourceApplyKerasMomentum.Options | useLocking (ブール型 useLocking) |
静的ResourceApplyKerasMomentum.Options | useNesterov (ブール型 useNesterov) |
継承されたメソッド
パブリックメソッド
public static ResourceApplyKerasMomentum create (スコープscope,オペランド<?>var,オペランド<?>accum,オペランド<T>lr,オペランド<T>grad,オペランド<T>モメンタム,オプション...オプション)
新しい ResourceApplyKerasMomentum オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
変数 | Variable() から取得する必要があります。 |
蓄積 | Variable() から取得する必要があります。 |
lr | スケーリング係数。スカラーでなければなりません。 |
卒業生 | グラデーション。 |
勢い | 勢い。スカラーでなければなりません。 |
オプション | オプションの属性値を持ちます |
戻り値
- ResourceApplyKerasMomentum の新しいインスタンス
public static ResourceApplyKerasMomentum.Options useLocking (Boolean useLocking)
パラメーター
使用ロック | 「True」の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 |
---|
public static ResourceApplyKerasMomentum.Options useNesterov (Boolean useNesterov)
パラメーター
使用ネステロフ | 「True」の場合、grad の計算に渡されるテンソルは var + moment * accum となるため、最終的に取得する var は実際には var + moment * accum になります。 |
---|