def step(in_t, h_tm1): h_t = lstm(in_t, h_tm1, [n_hid], n_hid, name="lstm_l1", random_state=random_state) return h_t
def step(in_t, h_tm1): h_t = lstm(in_t, h_tm1, n_hid, name="rec", random_state=random_state) return h_t
def step(in_t, h_tm1): h_t = lstm(in_t, h_tm1, [n_in], n_hid, name=None, random_state=random_state) return h_t