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