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