From aaa54b10b37d03a0a02e717e497512b3fadd9179 Mon Sep 17 00:00:00 2001
From: rasbt <mail@sebastianraschka.com>
Date: Wed, 12 Jun 2024 19:43:25 -0500
Subject: [PATCH] dim-consistency

---
 ch03/01_main-chapter-code/ch03.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ch03/01_main-chapter-code/ch03.ipynb b/ch03/01_main-chapter-code/ch03.ipynb
index 1ba30a9..12019b5 100644
--- a/ch03/01_main-chapter-code/ch03.ipynb
+++ b/ch03/01_main-chapter-code/ch03.ipynb
@@ -1369,7 +1369,7 @@
     }
    ],
    "source": [
-    "row_sums = masked_simple.sum(dim=1, keepdim=True)\n",
+    "row_sums = masked_simple.sum(dim=-1, keepdim=True)\n",
     "masked_simple_norm = masked_simple / row_sums\n",
     "print(masked_simple_norm)"
    ]
@@ -2027,7 +2027,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.6"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,