rearrange flatten layer

Author: dsilver829

LeNet-Lab-Solution.ipynb

@@ -197,9 +197,9 @@
     "\n",
     "    # SOLUTION: Flatten Layer.\n",
     "    fc1 = flatten(conv2)\n",
+    "    fc1_shape = (fc1.get_shape().as_list()[-1], 120)\n",
     "    \n",
     "    # SOLUTION: Fully Connected Layer 1. Input = 5x5x16. Output = 120.\n",
-    "    fc1_shape = (fc1.get_shape().as_list()[-1], 120)\n",
     "    fc1_W     = tf.Variable(tf.truncated_normal(shape=(fc1_shape)))\n",
     "    fc1_b     = tf.Variable(tf.zeros(120))\n",
     "    fc1       = tf.matmul(fc1, fc1_W) + fc1_b\n",