Merge pull request fastai#136 from appleby/lesson6-expand-dims

Use np.squeeze and np.atleast_3d to munge dimensions for {x,y}_rnn
deshion · Aug 2, 2017 · 9179eb4 · 9179eb4
2 parents b8f6645 + 9ab8276
commit 9179eb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeplearning1/nbs/lesson6.ipynb b/deeplearning1/nbs/lesson6.ipynb
@@ -2171,8 +2171,8 @@
    },
    "outputs": [],
    "source": [
-    "x_rnn=np.stack(xs, axis=1)\n",
-    "y_rnn=np.expand_dims(np.stack(ys, axis=1), -1)"
+    "x_rnn=np.stack(np.squeeze(xs), axis=1)\n",
+    "y_rnn=np.atleast_3d(np.stack(ys, axis=1))"
    ]
   },
   {