מחשב את התפשטות תא GRU קדימה עבור שלב זמן אחד.
Args x: קלט לתא GRU. h_prev: קלט מצב מתא GRU הקודם. w_ru: מטריצת משקל עבור שער האיפוס והעדכון. w_c: מטריצת משקל לשער חיבור התא. b_ru: וקטור הטיה עבור שער האיפוס והעדכון. b_c: וקטור הטיה עבור שער חיבור התא.
מחזירה r: פלט של שער האיפוס. u: פלט של שער העדכון. ג: פלט של שער חיבור התא. h: מצב נוכחי של תא GRU.
הערה לגבי סימון המשתנים:
שרשור של a ו-b מיוצג על ידי a_b מכפלת נקודה מבחינת היסודות של a ו-b מיוצגת על ידי ab מכפלת נקודה מבחינה אלמנט מיוצגת על ידי \circ כפל מטריקס מיוצג על ידי *
הטיות מאתחלות עם: `b_ru` - constant_initializer(1.0) `b_c` - constant_initializer(0.0)
קרנל אופ זה מיישם את המשוואות המתמטיות הבאות:
x_h_prev = [x, h_prev]
[r_bar u_bar] = x_h_prev * w_ru + b_ru
r = sigmoid(r_bar)
u = sigmoid(u_bar)
h_prevr = h_prev \circ r
x_h_prevr = [x h_prevr]
c_bar = x_h_prevr * w_c + b_c
c = tanh(c_bar)
h = (1-u) \circ c + u \circ h_prev
שיטות ציבוריות
פלט <T> | ג () |
סטטי <T מרחיב מספר> GRUBlockCell <T> | |
פלט <T> | ח () |
פלט <T> | r () |
פלט <T> | u () |
שיטות בירושה
שיטות ציבוריות
STATIC GRUBlockCell <T> create ( scope scope, Operand <T> x, Operand <T> hPrev, Operand <T> wRu, Operand <T> wC, Operand <T> bRu, Operand <T> bC)
שיטת מפעל ליצירת מחלקה העוטפת פעולת GRUBlockCell חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
מחזיר
- מופע חדש של GRUBlockCell