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)

קרנל אופ זה מיישם את המשוואות המתמטיות הבאות:

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