use torch.softmax instead of F.softmax

2026-04-10 12:33:42 +00:00 · 2024-01-28 12:25:49 -06:00
parent eb142491b2
commit 22442b7012
1 changed files with 2 additions and 2 deletions
--- a/appendix-A/03_main-chapter-code/code-part1.ipynb
+++ b/appendix-A/03_main-chapter-code/code-part1.ipynb
@@ -700,7 +700,7 @@
   ],
   "source": [
    "with torch.no_grad():\n",
-    "    out = F.softmax(model(X), dim=1)\n",
+    "    out = torch.softmax(model(X), dim=1)\n",
    "print(out)"
   ]
  },
@@ -986,7 +986,7 @@
   ],
   "source": [
    "torch.set_printoptions(sci_mode=False)\n",
-    "probas = F.softmax(outputs, dim=1)\n",
+    "probas = torch.softmax(outputs, dim=1)\n",
    "print(probas)\n",
    "\n",
    "predictions = torch.argmax(outputs, dim=1)\n",