Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
tomsanbear committed Mar 12, 2024
1 parent 436883e commit 2fbe6f8
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@

# candle-einops

This library is a fork from the original library that was implemented and based ont he TCH library. It
This library is a fork of [einops](https://github.com/VasanthakumarV/einops) intended to bring support for einops to [Candle](https://github.com/huggingface/candle)

Difference from the python version,
Difference from the python version:

- All code generated at compile time, avoiding the need for caching
- One common api for rearrange, reduce and repeat operations
Expand Down Expand Up @@ -68,7 +68,6 @@ let output = einops!("b h (w w2:2) c -> (h w2) (b w) c", &input);
```

__Reduce__
Note: the prod operation is not implemented until candle upstream deals with merging it in.

We can reduce axes using operations like, `sum`, `min`, `max`, and `mean`.
if the same operations has to be performed on multiple continuous axes we can do `sum(a b c)`
Expand Down

0 comments on commit 2fbe6f8

Please sign in to comment.