ResourceSparseApplyMomentum

מחלקה אחרונה ציבורית 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.