tensorflow:: ops:: ApplyMomentum

モーメンタムスキームに従って「*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 になります。

返品:

• Output : "var" と同じです。

公開属性

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 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs UseLocking(
  bool x
)

Attrs UseNesterov(
  bool x
)