diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index f89bc451279785072598f658690037a83908718b..3fcc7367bc7e05c071e6f3ef3fab6c204ee15e4e 100644 --- a/arraycontext/impl/pytato/fake_numpy.py +++ b/arraycontext/impl/pytato/fake_numpy.py @@ -107,6 +107,9 @@ class PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace): lambda *args: pt.stack(arrays=args, axis=axis), *arrays) + def broadcast_to(self, array, shape): + return rec_map_array_container(partial(pt.broadcast_to, shape=shape), array) + # {{{ relational operators def equal(self, x, y):