specify contiguous before flattening reconstructions

cezannec · cezannec · commit 607b7ab12daa · 2020-05-15T19:04:22.000-07:00
diff --git a/Capsule_Network.ipynb b/Capsule_Network.ipynb
@@ -620,11 +620,7 @@
     "        # create reconstructed pixels\n",
     "        x = x * y[:, :, None]\n",
     "        # flatten image into a vector shape (batch_size, vector_dim)\n",
-<<<<<<< HEAD
     "        flattened_x = x.contiguous().view(x.size(0), -1)\n",
-=======
-    "        flattened_x = x.reshape(x.size(0), -1)\n",
->>>>>>> 3818402479096260941d04b79061f35cc262aaa2
     "        # create reconstructed image vectors\n",
     "        reconstructions = self.linear_layers(flattened_x)\n",
     "        \n",
