diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 4dad159fd90e794c258816a6b35db3cf7a912121..f2e3d2e8e689baa8f2f7568d221f8bb8b9d958a0 100644 --- a/arraycontext/impl/pytato/fake_numpy.py +++ b/arraycontext/impl/pytato/fake_numpy.py @@ -73,7 +73,8 @@ class PytatoFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace): def zeros_like(self, ary): def _zeros_like(array): - return self._array_context.zeros(array.shape, array.dtype) + return self._array_context.zeros( + array.shape, array.dtype).copy(axes=array.axes, tags=array.tags) return self._array_context._rec_map_container( _zeros_like, ary, default_scalar=0) @@ -83,7 +84,8 @@ class PytatoFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace): def full_like(self, ary, fill_value): def _full_like(subary): - return pt.full(subary.shape, fill_value, subary.dtype) + return pt.full(subary.shape, fill_value, subary.dtype).copy( + axes=subary.axes, tags=subary.tags) return self._array_context._rec_map_container( _full_like, ary, default_scalar=fill_value)