Fix order of arguments in dot product

aesara-devs · Sep 1, 2021 · d321748 · d321748
1 parent 48c3b98
commit d321748
Show file tree

Hide file tree

Showing 2 changed files with 97 additions and 43 deletions.
diff --git a/aehmc/metrics.py b/aehmc/metrics.py
@@ -64,7 +64,7 @@ def gaussian_metric(
 
     def momentum_generator(srng: RandomStream) -> TensorVariable:
         norm_samples = srng.normal(0, 1, size=shape)
-        momentum = dot(norm_samples, mass_matrix_sqrt)
+        momentum = dot(mass_matrix_sqrt, norm_samples)
         return momentum
 
     def kinetic_energy(momentum: TensorVariable) -> TensorVariable: