เทนเซอร์โฟลว์:: ปฏิบัติการ:: สมัครอดัม
#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+)
ข้อโต้แย้ง:
- ขอบเขต: วัตถุ ขอบเขต
- var: ควรมาจากตัวแปร ()
- m: ควรมาจากตัวแปร ()
- v: ควรมาจากตัวแปร ()
- beta1_power: ต้องเป็นสเกลาร์
- beta2_power: ต้องเป็นสเกลาร์
- lr: ปัจจัยการปรับขนาด ต้องเป็นสเกลาร์
- beta1: ปัจจัยโมเมนตัม ต้องเป็นสเกลาร์
- beta2: ปัจจัยโมเมนตัม ต้องเป็นสเกลาร์
- เอปไซลอน: ระยะสัน ต้องเป็นสเกลาร์
- ผู้สำเร็จการศึกษา: การไล่ระดับสี
แอ็ตทริบิวต์ทางเลือก (ดู Attrs
):
- use_locking: หากเป็น
True
การอัปเดตเทนเซอร์ var, m และ v จะได้รับการปกป้องด้วยการล็อค มิฉะนั้นพฤติกรรมจะไม่ได้กำหนดไว้ แต่อาจแสดงความขัดแย้งน้อยลง - use_nesterov: หากเป็น
True
ให้ใช้การอัปเดต Nesterov
ผลตอบแทน:
-
Output
: เหมือนกับ "var"
ตัวสร้างและผู้ทำลาย | |
---|---|
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) |
โครงสร้าง | |
---|---|
เทนเซอร์โฟลว์ :: 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 )