Skip to content

Commit

Permalink
Add dict syntax to the einops layers example (#342)
Browse files Browse the repository at this point in the history
* Add dict syntax to the einops layers example

* Update 2-einops-for-deep-learning.ipynb

* Update 2-einops-for-deep-learning.ipynb

* Update docs/2-einops-for-deep-learning.ipynb

Co-authored-by: Alex Rogozhnikov <[email protected]>

---------

Co-authored-by: Alex Rogozhnikov <[email protected]>
eadadi and arogozhnikov authored Sep 21, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent 6113d02 commit 8112f83
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docs/2-einops-for-deep-learning.ipynb
Original file line number Diff line number Diff line change
@@ -1161,6 +1161,8 @@
" Linear(16*5*5, 120), \n",
" ReLU(),\n",
" Linear(120, 10), \n",
" # In flax, the {'axis': value} syntax for specifying values for axes is mandatory:\n",
" # Rearrange('(b1 b2) d -> b1 b2 d', {'b1': 12}), \n",
")\n",
"```"
]

0 comments on commit 8112f83

Please sign in to comment.