books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

dim-consistency

Browse Source
This commit is contained in:
rasbt
2024-06-12 19:43:25 -05:00
parent bcccda728b
commit aaa54b10b3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ch03/01_main-chapter-code/ch03.ipynb
View File

@@ -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,