Google I/O एक लपेट है! TensorFlow सत्रों पर पकड़ बनाएं सत्र देखें

टेंसरफ़्लो :: ऑप्स :: 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 

सार्वजनिक स्थैतिक कार्य

उपयोग करना

02 फीए 2 ए 10

UseNesterov

Attrs UseNesterov(
  bool x
)