टेंसरफ़्लो :: ऑप्स :: SparseApplyMomentum
#include <training_ops.h>
गति योजना के अनुसार '* var' और '* एकत्रित' में प्रासंगिक प्रविष्टियाँ अद्यतन करें।
सारांश
यदि आप Nesterov गति का उपयोग करना चाहते हैं तो use_nesterov = True सेट करें।
हम उन पंक्तियों के लिए हैं जिनके लिए हम वर्गीकृत हैं, हम निम्नानुसार var और संचित को अपडेट करते हैं:
$$accum = accum * momentum + grad$$ $$var -= lr * accum$$
तर्क:
- गुंजाइश: एक स्कोप ऑब्जेक्ट
- var: एक चर () से होना चाहिए।
- संचित: एक चर () से होना चाहिए।
- lr: सीखने की दर। एक स्केलर होना चाहिए।
- grad: ढाल।
- सूचकांक: संस्करण और संचय के पहले आयाम में सूचकांकों का एक वेक्टर।
- गति: गति। एक स्केलर होना चाहिए।
वैकल्पिक विशेषताएँ ( Attrs
देखें):
- use_locking: यदि
True
, तो var का अपडेट और संचित टेंसरों को लॉक द्वारा संरक्षित किया जाएगा; अन्यथा व्यवहार अपरिभाषित है, लेकिन कम विवाद को प्रदर्शित कर सकता है। - use_nesterov: यदि
True
, तो टेनर को गणना करने के लिए पास किया गया var - lr * momentum * संचित होगा, इसलिए अंत में, आपको जो var मिलेगा वह वास्तव में var - lr * momentum * संचित है।
रिटर्न:
-
Output
: "var" के समान।
कंस्ट्रक्टर और डिस्ट्रक्टर्स | |
---|---|
SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const SparseApplyMomentum::Attrs & attrs) |
सार्वजनिक विशेषताएँ | |
---|---|
operation | |
out |
सार्वजनिक कार्य | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
सार्वजनिक स्थैतिक कार्य | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
संरचनाएं | |
---|---|
टेंसोफ़्लो :: ऑप्स :: स्पार्सेप्लीमोमेंटम :: एट्र्स | वैकल्पिक विशेषता SparseApplyMomentum के लिए बसता है । |
सार्वजनिक विशेषताएँ
ऑपरेशन
Operation operation
बाहर
::tensorflow::Output out
सार्वजनिक कार्य
SparseApplyMomentum
SparseApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum )
SparseApplyMomentum
SparseApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input indices, ::tensorflow::Input momentum, const SparseApplyMomentum::Attrs & attrs )
नोड
::tensorflow::Node * node() const
ऑपरेटर :: टेंसरफ़्लो :: इनपुट
operator::tensorflow::Input() constहै
ऑपरेटर :: टेंसरफ़्लो :: आउटपुट
operator::tensorflow::Output() const
सार्वजनिक स्थैतिक कार्य
उपयोग करना
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )