From adc68cb1cf6ca8a2fb084f5a0b01a526cd127119 Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 25 Jan 2021 16:28:52 +0100 Subject: [PATCH] add support for complex64 tensors --- Example.ipynb | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/Example.ipynb b/Example.ipynb index df49454..518f6a3 100755 --- a/Example.ipynb +++ b/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" ] } ],