Skip to content

Commit

Permalink
Fix some expressions in tutorial
Browse files Browse the repository at this point in the history
  • Loading branch information
fferflo committed Feb 13, 2025
1 parent b7a90b4 commit dab9ab1
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions docs/source/gettingstarted/tutorial_notation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -289,8 +289,7 @@ Some other examples:

.. code::
einx.flip("a [b]", x, c=2) # Flip pairs of values
einx.add("... [c]", x, b) # Add bias
einx.flip("a [b]", x, b=2) # Flip pairs of values
einx.get_at("b [h w] c, b i [2] -> b i c", x, indices) # Gather values
einx.softmax("b q [k] h", attn) # Part of attention operation
Expand Down
2 changes: 1 addition & 1 deletion docs/source/gettingstarted/tutorial_ops.rst
Original file line number Diff line number Diff line change
Expand Up @@ -240,7 +240,7 @@ All operations support expression rearranging:

.. code::
einx.add_at("b ([h w]) c, ([2] b) i, c i -> c [h w] b", image, coordinates, updates)
einx.add_at("b ([h w]) c, ([2] b) i, c i -> c [h w] b", image, coordinates, updates, h=256)
Several operations from classical tensor frameworks can be expressed using universal einx indexing functions:

Expand Down

0 comments on commit dab9ab1

Please sign in to comment.