כיתת גמר ציבורית ResourceApplyMomentum
עדכן את '*var' לפי סכימת המומנטום.
הגדר use_nesterov = True אם אתה רוצה להשתמש במומנטום Nesterov.
accum = accum * מומנטום + grad var -= lr * accum
כיתות מקוננות
| מעמד | ResourceApplyMomentum.Options | תכונות אופציונליות עבור ResourceApplyMomentum | |
קבועים
| חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
| סטטי <T מרחיב את TType > ResourceApplyMomentum | |
| static ResourceApplyMomentum.Options | useLocking (useLocking בוליאני) |
| static ResourceApplyMomentum.Options | useNesterov (בוליאני useNesterov) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "ResourceApplyMomentum"
שיטות ציבוריות
public static ResourceApplyMomentum create ( scope scope, Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <T> מומנטום, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת ResourceApplyMomentum חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| var | צריך להיות ממשתנה(). |
| לצבור | צריך להיות ממשתנה(). |
| lr | גורם קנה מידה. חייב להיות סקלר. |
| גראד | השיפוע. |
| תְנוּפָה | תְנוּפָה. חייב להיות סקלר. |
| אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של ResourceApplyMomentum
public static ResourceApplyMomentum.Options useLocking (שימוש בוליאני)
פרמטרים
| השתמש בנעילה | אם 'נכון', עדכון טנסור ה- var ו-acum יהיה מוגן על ידי מנעול; אחרת ההתנהגות אינה מוגדרת, אך עלולה להפגין פחות מחלוקת. |
|---|
public static ResourceApplyMomentum.Options useNesterov (בוליאני useNesterov)
פרמטרים
| השתמש ב-Nesterov | אם 'נכון', הטנזור שיעבור לחישוב גראד יהיה var - lr * מומנטום * accum, אז בסופו של דבר, ה-var שאתה מקבל הוא למעשה var - lr * מומנטום * accum. |
|---|