yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-06-26 23:50:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

use less ambiguous var name

Browse Source
This commit is contained in:
rasbt 2024-02-13 07:05:37 -06:00
parent 320f63829f
commit 231a854ae7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ch04/01_main-chapter-code/ch04.ipynb
View File

@ -301,10 +301,10 @@
"torch.manual_seed(123)\n", "torch.manual_seed(123)\n",
"\n", "\n",
"# create 2 training examples with 5 dimensions (features) each\n", "# create 2 training examples with 5 dimensions (features) each\n",
"batch_input = torch.randn(2, 5) \n", "batch_example = torch.randn(2, 5) \n",
"\n", "\n",
"layer = nn.Sequential(nn.Linear(5, 6), nn.ReLU())\n", "layer = nn.Sequential(nn.Linear(5, 6), nn.ReLU())\n",
"out = layer(batch_input)\n", "out = layer(batch_example)\n",
"print(out)" "print(out)"
] ]
}, },
@ -492,7 +492,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"ln = LayerNorm(emb_dim=5)\n", "ln = LayerNorm(emb_dim=5)\n",
"out_ln = ln(batch_input)" "out_ln = ln(batch_example)"
] ]
}, },
{ {