tf.contrib.rnn.GRUBlockCell

class tf.contrib.rnn.GRUBlockCell

Block GRU cell implementation.

The implementation is based on: http://arxiv.org/abs/1406.1078 Computes the LSTM cell forward propagation for 1 time step.

This kernel op implements the following mathematical equations:

Baises are initialized with : 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 ```