From 8c4a371c8b3785f5349c5f9f72238a38cbf378b3 Mon Sep 17 00:00:00 2001 From: Colin Carroll Date: Sun, 8 Oct 2017 16:42:11 -0400 Subject: [PATCH] Allow reasonable sampling from Normal distribution --- pymc3/distributions/continuous.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pymc3/distributions/continuous.py b/pymc3/distributions/continuous.py index 7dbc366943..b2a4fa95f9 100644 --- a/pymc3/distributions/continuous.py +++ b/pymc3/distributions/continuous.py @@ -284,6 +284,7 @@ def __init__(self, mu=0, sd=None, tau=None, **kwargs): assert_negative_support(tau, 'tau', 'Normal') super(Normal, self).__init__(**kwargs) + self.shape = np.broadcast(mu, sd).shape def random(self, point=None, size=None, repeat=None): mu, tau, _ = draw_values([self.mu, self.tau, self.sd],