tensorflow:: ops:: ResourceApplyMomentum
#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の場合、grad を計算するために渡されるテンソルは var - lr * momentum * accum になります。したがって、最終的に得られる var は実際には var - lr * momentum * accum になります。
返品:
- 作成された
Operation
構築者と破壊者 | |
|---|---|
ResourceApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum) | |
ResourceApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum, const ResourceApplyMomentum::Attrs & attrs) |
公開属性 | |
|---|---|
operation | |
公的機能 | |
|---|---|
operator::tensorflow::Operation () const | |
パブリック静的関数 | |
|---|---|
UseLocking (bool x) | |
UseNesterov (bool x) | |
構造 | |
|---|---|
| tensorflow:: ops:: ResourceApplyMomentum:: Attrs | ResourceApplyMomentumのオプション属性設定子。 |
公開属性
手術
Operation operation
公的機能
リソース適用モメンタム
ResourceApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
リソース適用モメンタム
ResourceApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ResourceApplyMomentum::Attrs & attrs )
オペレーター::tensorflow::Operation
operator::tensorflow::Operation() const
パブリック静的関数
ロックを使用する
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )