Calcula la propagación hacia adelante de la celda GRU para 1 paso de tiempo.
Args x: Entrada a la celda GRU. h_prev: Entrada de estado de la celda GRU anterior. w_ru: Matriz de peso para la puerta de reinicio y actualización. w_c: Matriz de pesos para la puerta de conexión de celdas. b_ru: Vector de polarización para la puerta de reinicio y actualización. b_c: Vector de polarización para la puerta de conexión de la celda.
Devuelve r: Salida de la puerta de reinicio. u: Salida de la puerta de actualización. c: Salida de la puerta de conexión de la celda. h: Estado actual de la celda GRU.
Nota sobre la notación de las variables:
La concatenación de a y b está representada por a_b El producto escalar por elementos de a y b está representado por ab El producto escalar por elementos está representado por \circ La multiplicación de matrices está representada por *
Los sesgos se inicializan con: `b_ru` - constant_initializer(1.0) `b_c` - constant_initializer(0.0)
Esta operación del kernel implementa las siguientes ecuaciones matemáticas:
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
Métodos públicos
| Salida <T> | do () |
| estático <T extiende Número> GRUBlockCell <T> | |
| Salida <T> | h () |
| Salida <T> | r () |
| Salida <T> | tu () |
Métodos Heredados
Métodos públicos
GRUBlockCell <T> estático público crear ( Alcance alcance, Operando <T> x, Operando <T> hPrev, Operando <T> wRu, Operando <T> wC, Operando <T> bRu, Operando <T> bC)
Método de fábrica para crear una clase que envuelve una nueva operación GRUBlockCell.
Parámetros
| alcance | alcance actual |
|---|
Devoluciones
- una nueva instancia de GRUBlockCell