Fixed truncated-normal bug

AzizZayed · Jun 15, 2021 · 0092f8e · 0092f8e
1 parent a6ded8c
commit 0092f8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/java/ai/djl/ndarray/BaseNDManager.java b/api/src/main/java/ai/djl/ndarray/BaseNDManager.java
@@ -169,7 +169,7 @@ public NDArray truncatedNormal(float loc, float scale, Shape shape, DataType dat
             dist[i] = sample;
         }
 
-        return create(dist).addi(loc).muli(scale).reshape(shape).toType(dataType, false);
+        return create(dist).muli(scale).addi(loc).reshape(shape).toType(dataType, false);
     }
 
     /** {@inheritDoc} */