टेंसरफ़्लो:: ऑप्स:: मोमेंटम लागू करें

#include <training_ops.h>

गति योजना के अनुसार '*var' को अपडेट करें।

सारांश

यदि आप नेस्टरोव संवेग का उपयोग करना चाहते हैं तो use_nesterov = True सेट करें।

संचय = संचय * संवेग + ग्रेड वर - = एलआर * संचय

तर्क:

  • स्कोप: एक स्कोप ऑब्जेक्ट
  • 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
)