SBVM sampler fix (#3261)

pyro-ppl · Aug 25, 2023 · cc8e545 · cc8e545
1 parent 1048a8b
commit cc8e545
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/distributions/sine_bivariate_von_mises.py b/pyro/distributions/sine_bivariate_von_mises.py
@@ -204,7 +204,7 @@ def sample(self, sample_shape=torch.Size()):
             curr_b0 = b0[missing > 0]
 
             x = (
-                torch.distributions.Normal(0.0, torch.sqrt(1 + 2 * curr_eig / curr_b0))
+                torch.distributions.Normal(0.0, torch.rsqrt(1 + 2 * curr_eig / curr_b0))
                 .sample((missing[missing > 0].min(),))
                 .view(2, -1, missing[missing > 0].min())
             )