Fixes #1

chapter5/GCN_Cora.ipynb

@@ -399,7 +399,7 @@
     "        test_mask_logits = logits[mask]\n",
     "        predict_y = test_mask_logits.max(1)[1]\n",
     "        accuarcy = torch.eq(predict_y, tensor_y[mask]).float().mean()\n",
-    "    return accuarcy, test_mask_logits.numpy(), tensor_y[mask].numpy()\n"
+    "    return accuarcy, test_mask_logits.cpu().numpy(), tensor_y[mask].cpu().numpy()\n"
    ]
   },
   {