From d179e07e7de0f8e96d58b0f0b6b6fc8c21c4fd74 Mon Sep 17 00:00:00 2001
From: Florian Fervers <florian.fervers@gmail.com>
Date: Tue, 23 Jan 2024 17:20:24 +0100
Subject: [PATCH] Update readme

---
 readme.md | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/readme.md b/readme.md
index f51a127..1152a30 100644
--- a/readme.md
+++ b/readme.md
@@ -113,12 +113,11 @@ einx traces the required backend operations for a given call into graph represen
 >>> x = np.zeros((3, 10, 10))
 >>> graph = einx.sum("... (g [c])", x, g=2, graph=True)
 >>> print(graph)
-
-def reduce(i0, backend):
-    x1 = backend.to_tensor(i0)
-    x2 = backend.reshape(x1, (3, 10, 2, 5))
-    x3 = backend.sum(x2, axis=3)
-    return x3
+# backend: einx.backend.numpy
+def op0(i0):
+    x1 = backend.reshape(i0, (3, 10, 2, 5))
+    x0 = backend.sum(x1, axis=3)
+    return x0
 ```
 
 See [Just-in-time compilation](https://einx.readthedocs.io/en/latest/gettingstarted/jit.html) for more details.
\ No newline at end of file