SparseApplyMomentum.Options

パブリック静的クラスSparseApplyMomentum.Options

SparseApplyMomentumのオプションの属性

パブリックメソッド

SparseApplyMomentum.Options
useLocking (ブール型 useLocking)
SparseApplyMomentum.Options
useNesterov (ブール型 useNesterov)

継承されたメソッド

パブリックメソッド

public SparseApplyMomentum.Options useLocking (ブール型 useLocking)

パラメーター
使用ロック「True」の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。

public SparseApplyMomentum.Options useNesterov (ブール型 useNesterov)

パラメーター
使用ネステロフ「True」の場合、grad を計算するために渡されるテンソルは var - lr *momentum *accum となるため、最終的に取得する var は実際には var - lr *momentum *accum になります。