टेंसरफ़्लो:: ऑप्स:: मोमेंटम लागू करें
#include <training_ops.h>
गति योजना के अनुसार '*var' को अपडेट करें।
सारांश
यदि आप उपयोग_नेस्टरोव = सत्य सेट करें
नेस्टरोव गति का उपयोग करना चाहते हैं।
संचय = संचय * संवेग + ग्रेड वर - = एलआर * संचय
तर्क:
- स्कोप: एक स्कोप ऑब्जेक्ट
- var: एक वेरिएबल() से होना चाहिए।
- संचय: एक वेरिएबल() से होना चाहिए।
- एलआर: स्केलिंग कारक। एक अदिश राशि होनी चाहिए.
- ग्रेड: ग्रेडिएंट.
- संवेग: संवेग. एक अदिश राशि होनी चाहिए.
वैकल्पिक विशेषताएँ (देखें Attrs
):
- उपयोग_लॉकिंग: यदि
True
, तो var और Accum Tensors का अद्यतनीकरण लॉक द्वारा संरक्षित किया जाएगा; अन्यथा व्यवहार अपरिभाषित है, लेकिन कम विवाद प्रदर्शित कर सकता है। - उपयोग_नेस्टरोव: यदि
True
, तो ग्रेड की गणना करने के लिए पारित टेंसर var - lr * संवेग * accum होगा, इसलिए अंत में, आपको जो var मिलता है वह वास्तव में var - lr * संवेग * accum है।
रिटर्न:
-
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 )
नेस्टरोव का उपयोग करें
Attrs UseNesterov( bool x )