GRUBlockCell

שיעור הגמר הציבורי GRUBlockCell

מחשב את התפשטות תא 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)

Kernel op זה מיישם את המשוואות המתמטיות הבאות:

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>
ליצור ( scope scope, Operand <T> x, Operand <T> hPrev, Operand <T> wRu, Operand <T> wC, Operand <T> bRu, Operand <T> bC)
שיטת מפעל ליצירת מחלקה העוטפת פעולת GRUBlockCell חדשה.
פלט <T>
ח ()
פלט <T>
r ()
פלט <T>
u ()

שיטות בירושה

שיטות ציבוריות

פלט ציבורי <T> c ()

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

פלט ציבורי <T> h ()

פלט ציבורי <T> r ()

פלט ציבורי <T> u ()