remove no_grad on loc and scale for normal dist

will-maclean · Sep 8, 2024 · e8a5e2d · e8a5e2d
1 parent 49b5dce
commit e8a5e2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/common/distributions/normal.rs b/src/common/distributions/normal.rs
@@ -24,8 +24,8 @@ impl<B: Backend, const D: usize> Normal<B, D> {
         );
 
         Self {
-            loc: loc.no_grad(),
-            scale: scale.no_grad(),
+            loc,
+            scale,
         }
     }
 }