diff --git a/arraycontext/impl/numpy/fake_numpy.py b/arraycontext/impl/numpy/fake_numpy.py index 4ac10055e5143b0f8f14030718eb2831fa853dd0..b7a2335a1a50ca58a235e180b66c264e619ebabd 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