Fixes to MUT1 layer + test.

lmjohns3 · Aug 13, 2015 · c151f1a · c151f1a
1 parent 0f55784
commit c151f1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/test/layers_test.py b/test/layers_test.py
@@ -293,7 +293,7 @@ def test_create(self):
         self.assert_param_names(['bh', 'br', 'bz',
                                  'hh', 'hr',
                                  'xh', 'xr', 'xz'])
-        self.assert_count(84)
+        self.assert_count(68)
 
     def test_transform(self):
         out, upd = self.l.transform(dict(out=self.x))

diff --git a/theanets/layers/recurrent.py b/theanets/layers/recurrent.py
@@ -841,9 +841,8 @@ def fn(x_t, r_t, z_t, h_tm1):
         (p, h, o), updates = self._scan(
             fn,
             [TT.tanh(TT.dot(x, self.find('xh'))) + self.find('bh'),
-             TT.dot(x, self.find('xr')) + self.find('br'),
-             z],
-            [None, None, None, x])
+             TT.dot(x, self.find('xr')) + self.find('br'), z],
+            [None, None, x])
 
         # output is:  (time, batch, output)
         # we want:    (batch, time, output)