yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-12-16 09:39:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

use .shape instead of .size() for consistency

Browse Source
This commit is contained in:
rasbt 2024-02-25 08:47:25 -06:00
parent cdcd73ba7f
commit 3f186ab072
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -756,7 +756,7 @@
" # Compute the output of the current layer\n",
" layer_output = layer(x)\n",
" # Check if shortcut can be applied\n",
" if self.use_shortcut and x.size() == layer_output.size():\n",
" if self.use_shortcut and x.shape == layer_output.shape:\n",
" x = x + layer_output\n",
" else:\n",
" x = layer_output\n",