MLコミュニティデーは11月9日です! TensorFlow、JAXからの更新のために私たちに参加し、より多くの詳細をご覧ください

tensorflow :: ops :: SparseApplyMomentum :: Attrs

#include <training_ops.h>

SparseApplyMomentumのオプションの属性セッター。

概要

パブリック属性

use_locking_ = false
bool
use_nesterov_ = false
bool

公の行事

UseLocking (bool x)
TF_MUST_USE_RESULT Attrs
True場合、varテンソルとaccumテンソルの更新はロックによって保護されます。それ以外の場合、動作は定義されていませんが、競合が少なくなる可能性があります。
UseNesterov (bool x)
TF_MUST_USE_RESULT Attrs
True場合、gradを計算するために渡されるテンソルはvar --lr * momentum * accumになるため、最終的に取得するvarは実際にはvar --lr * momentum * accumになります。

パブリック属性

use_locking_

bool tensorflow::ops::SparseApplyMomentum::Attrs::use_locking_ = false

use_nesterov_

bool tensorflow::ops::SparseApplyMomentum::Attrs::use_nesterov_ = false

公の行事

UseLocking

TF_MUST_USE_RESULT Attrs tensorflow::ops::SparseApplyMomentum::Attrs::UseLocking(
  bool x
)

True場合、varテンソルとaccumテンソルの更新はロックによって保護されます。それ以外の場合、動作は定義されていませんが、競合が少なくなる可能性があります。

デフォルトはfalse

UseNesterov

TF_MUST_USE_RESULT Attrs tensorflow::ops::SparseApplyMomentum::Attrs::UseNesterov(
  bool x
)

True場合、gradを計算するために渡されるテンソルはvar --lr * momentum * accumになるため、最終的に取得するvarは実際にはvar --lr * momentum * accumになります。

デフォルトはfalse