diff --git a/torchquad/integration/gaussian.py b/torchquad/integration/gaussian.py index 0d3c9612..17051681 100644 --- a/torchquad/integration/gaussian.py +++ b/torchquad/integration/gaussian.py @@ -65,7 +65,7 @@ def _weights(self, N, dim, backend, requires_grad=False): ).ravel() else: return anp.prod( - anp.meshgrid(*([weights] * dim), like=backend), axis=0 + anp.stack(anp.meshgrid(*([weights] * dim), like=backend)), axis=0 ).ravel() def _roots(self, N, backend, requires_grad=False):