From ee3100a11bdbe01d573640e89803d338aff947e1 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Tue, 20 Jun 2023 17:12:15 -0500
Subject: [PATCH] Numpy actx: add zeros_like, reshape
---
arraycontext/impl/numpy/fake_numpy.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/arraycontext/impl/numpy/fake_numpy.py b/arraycontext/impl/numpy/fake_numpy.py
index 4ac1005..b7a2335 100644
--- a/arraycontext/impl/numpy/fake_numpy.py
+++ b/arraycontext/impl/numpy/fake_numpy.py
@@ -146,4 +146,13 @@ class NumpyFakeNumpyNamespace(BaseFakeNumpyNamespace):
def linspace(self, *args, **kwargs):
return np.linspace(*args, **kwargs)
+ def zeros_like(self, ary):
+ return rec_map_array_container(np.zeros_like, ary)
+
+ def reshape(self, a, newshape, order="C"):
+ return rec_map_array_container(
+ lambda ary: ary.reshape(newshape, order=order),
+ a)
+
+
# vim: fdm=marker
--
GitLab