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