テンソルフロー::作戦::勢いを適用する
#include <training_ops.h>
運動量スキームに従って「*var」を更新します。
まとめ
次の場合は use_nesterov = True を設定します。
ネステロフの勢いを利用したい。
accum = accum * 勢い + grad var -= lr * accum
引数:
- スコープ:スコープオブジェクト
- var: Variable() から取得する必要があります。
- accum: Variable() から取得する必要があります。
- lr: スケーリング係数。スカラーでなければなりません。
- grad: グラデーション。
- 勢い:勢い。スカラーでなければなりません。
オプションの属性 ( Attrs
を参照):
- use_locking:
True
の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 - use_nesterov:
True
の場合、compute grad に渡されるテンソルは var - lr * momento * accum となるため、最終的に取得する var は実際には var - lr *勢い * accum になります。
戻り値:
-
Output
: 「var」と同じ。
コンストラクターとデストラクター | |
---|---|
ApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum) | |
ApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum, const ApplyMomentum::Attrs & attrs) |
パブリック属性 | |
---|---|
operation | |
out |
公共機能 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
パブリック静的関数 | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
構造体 | |
---|---|
tensorflow:: ops:: applyMomentum:: Attrs | applyMomentumのオプションの属性セッター。 |
パブリック属性
手術
Operation operation
外
::tensorflow::Output out
公共機能
勢いを適用する
ApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
勢いを適用する
ApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ApplyMomentum::Attrs & attrs )
ノード
::tensorflow::Node * node() const
演算子::tensorflow::入力
operator::tensorflow::Input() const
演算子::tensorflow::出力
operator::tensorflow::Output() const
パブリック静的関数
ロックを使用する
Attrs UseLocking( bool x )
ネステロフを使用する
Attrs UseNesterov( bool x )