تدفق التوتر:: العمليات:: قدم آدم
#include <training_ops.h>
قم بتحديث '*var' وفقًا لخوارزمية Adam.
ملخص
lrt:=learningrate∗1−betat2/(1−betat1)
mt:=beta1∗mt−1+(1−beta1)∗g
vt:=beta2∗vt−1+(1−beta2)∗g∗g
variable:=variable−lrt∗mt/(vt+)
الحجج:
- النطاق: كائن النطاق
- فار: يجب أن يكون من متغير ().
- م: يجب أن يكون من متغير ().
- v: يجب أن يكون من متغير ().
- beta1_power: يجب أن يكون عدديًا.
- beta2_power: يجب أن يكون عدديًا.
- lr: عامل التحجيم. يجب أن يكون العددية.
- beta1: عامل الزخم. يجب أن يكون العددية.
- beta2: عامل الزخم. يجب أن يكون العددية.
- إبسيلون: مصطلح ريدج. يجب أن يكون العددية.
- غراد: التدرج.
السمات الاختيارية (انظر Attrs
):
- use_locking: إذا كان
True
، فسيتم حماية تحديث الموترات var وm وv بواسطة قفل؛ وإلا فإن السلوك غير محدد، ولكنه قد يحمل قدرًا أقل من الخلاف. - use_nesterov: إذا كان
True
، يستخدم تحديث Nesterov.
العوائد:
-
Output
: نفس "فار".
البنائين والمدمرين | |
---|---|
ApplyAdam (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input v, :: tensorflow::Input beta1_power, :: tensorflow::Input beta2_power, :: tensorflow::Input lr, :: tensorflow::Input beta1, :: tensorflow::Input beta2, :: tensorflow::Input epsilon, :: tensorflow::Input grad) | |
ApplyAdam (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input v, :: tensorflow::Input beta1_power, :: tensorflow::Input beta2_power, :: tensorflow::Input lr, :: tensorflow::Input beta1, :: tensorflow::Input beta2, :: tensorflow::Input epsilon, :: tensorflow::Input grad, const ApplyAdam::Attrs & attrs) |
الوظائف العامة | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
وظائف ثابتة العامة | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
الهياكل | |
---|---|
Tensorflow:: ops:: ApplyAdam:: Attrs | محددات السمات الاختيارية لـ ApplyAdam . |
الصفات العامة
عملية
Operation operation
خارج
::tensorflow::Output out
الوظائف العامة
قدم آدم
ApplyAdam( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad )
قدم آدم
ApplyAdam( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad, const ApplyAdam::Attrs & attrs )