add support for complex64 tensors

Author: Your Name

Example.ipynb

@@ -86,7 +86,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
@@ -116,7 +116,23 @@
       "Train Epoch:   2 [     0/ 60000 (  0%)]\tLoss: 0.003121\n",
       "Train Epoch:   2 [  6400/ 60000 ( 11%)]\tLoss: 0.003100\n",
       "Train Epoch:   2 [ 12800/ 60000 ( 21%)]\tLoss: 0.001305\n",
-      "Train Epoch:   2 [ 19200/ 60000 ( 32%)]\tLoss: 0.017995\n"
+      "Train Epoch:   2 [ 19200/ 60000 ( 32%)]\tLoss: 0.017995\n",
+      "Train Epoch:   2 [ 25600/ 60000 ( 43%)]\tLoss: 0.116825\n",
+      "Train Epoch:   2 [ 32000/ 60000 ( 53%)]\tLoss: 0.018003\n",
+      "Train Epoch:   2 [ 38400/ 60000 ( 64%)]\tLoss: 0.002124\n",
+      "Train Epoch:   2 [ 44800/ 60000 ( 75%)]\tLoss: 0.001319\n",
+      "Train Epoch:   2 [ 51200/ 60000 ( 85%)]\tLoss: 0.000611\n",
+      "Train Epoch:   2 [ 57600/ 60000 ( 96%)]\tLoss: 0.013717\n",
+      "Train Epoch:   3 [     0/ 60000 (  0%)]\tLoss: 0.010614\n",
+      "Train Epoch:   3 [  6400/ 60000 ( 11%)]\tLoss: 0.003073\n",
+      "Train Epoch:   3 [ 12800/ 60000 ( 21%)]\tLoss: 0.004097\n",
+      "Train Epoch:   3 [ 19200/ 60000 ( 32%)]\tLoss: 0.001755\n",
+      "Train Epoch:   3 [ 25600/ 60000 ( 43%)]\tLoss: 0.005218\n",
+      "Train Epoch:   3 [ 32000/ 60000 ( 53%)]\tLoss: 0.003784\n",
+      "Train Epoch:   3 [ 38400/ 60000 ( 64%)]\tLoss: 0.011459\n",
+      "Train Epoch:   3 [ 44800/ 60000 ( 75%)]\tLoss: 0.022287\n",
+      "Train Epoch:   3 [ 51200/ 60000 ( 85%)]\tLoss: 0.031437\n",
+      "Train Epoch:   3 [ 57600/ 60000 ( 96%)]\tLoss: 0.001994\n"
      ]
     }
    ],