fix(example): fix example

jinmingyi1998 · Aug 6, 2023 · b591c25 · b591c25
1 parent 0436b6d
commit b591c25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/add.py b/examples/add.py
@@ -17,7 +17,7 @@ def add():
     r = Runner()
     a = np.random.random([64, 64])
     a = np.ascontiguousarray(a, dtype=np.float32)
-    b = np.random.random([64, 64, 64, 64])
+    b = np.random.random(a.shape)
     b = np.ascontiguousarray(b, dtype=np.float32)
 
     out = np.zeros_like(a)