yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-08-15 04:01:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ch03.ipynb

Browse Source
This commit is contained in:
Xiaotian Ma 2023-12-26 22:05:21 -06:00 committed by GitHub
parent 727615423d
commit c518adb0b7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -306,7 +306,7 @@
"attn_weights_2 = torch.softmax(attn_scores_2, dim=0)\n", "attn_weights_2 = torch.softmax(attn_scores_2, dim=0)\n",
"\n", "\n",
"print(\"Attention weights:\", attn_weights_2)\n", "print(\"Attention weights:\", attn_weights_2)\n",
"print(\"Sum:\", attn_weights.sum())" "print(\"Sum:\", attn_weights_2.sum())"
] ]
}, },
{ {