From 2fbe6f8c7519e5fd10efbb87f870fe19587c9b3f Mon Sep 17 00:00:00 2001 From: Thomas Santerre Date: Tue, 12 Mar 2024 15:59:04 -0400 Subject: [PATCH] update readme --- README.md | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 432730d..cd6da4c 100644 --- a/README.md +++ b/README.md @@ -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 @@ -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)`