パブリック最終クラスApplyMomentum
運動量スキームに従って「*var」を更新します。
Nesterov の勢いを使用したい場合は、use_nesterov = True を設定します。
accum = accum * 勢い + grad var -= lr * accum
ネストされたクラス
| クラス | モメンタムの適用オプション | ApplyMomentumのオプションの属性 | |
定数
| 弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
| 出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
| static <T extends TType > applyMomentum <T> | |
| 出力<T> | 外（） 「var」と同じです。 |
| 静的ApplyMomentum.Options | useLocking (ブール型 useLocking) |
| 静的ApplyMomentum.Options | useNesterov (ブール型 useNesterov) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "ApplyMomentum"
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static applyMomentum <T> create (スコープscope、オペランド<T> var、オペランド<T> accum、オペランド<T> lr、オペランド<T> grad、オペランド<T> モメンタム、オプション...オプション)
新しいApplyMomentumオペレーションをラップするクラスを作成するファクトリメソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 変数 | Variable() から取得する必要があります。 |
| 蓄積 | Variable() から取得する必要があります。 |
| lr | スケーリング係数。スカラーでなければなりません。 |
| 卒業生 | グラデーション。 |
| 勢い | 勢い。スカラーでなければなりません。 |
| オプション | オプションの属性値を持ちます |
戻り値
- applyMomentum の新しいインスタンス
public static applyMomentum.Options useLocking (ブール型 useLocking)
パラメーター
| 使用ロック | 「True」の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 |
|---|
public static applyMomentum.Options useNesterov (ブール型 useNesterov)
パラメーター
| 使用ネステロフ | 「True」の場合、grad を計算するために渡されるテンソルは var - lr *momentum *accum となるため、最終的に取得する var は実際には var - lr *momentum *accum になります。 |
|---|