chore: update README.md (tracel-ai#678)

hojjatabdollahi · Aug 23, 2023 · b60d931 · b60d931
1 parent 8f4faab
commit b60d931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -149,9 +149,9 @@ fn linear<B: Backend>(x: Tensor<B, 2>, weight: Tensor<B, 2>, bias: Tensor<B, 2>)
 fn main() {
     type Backend = NdArrayBackend<f32>;
 
-    let weight = Tensor::random([3, 3], Distribution::Standard);
+    let weight = Tensor::random([3, 3], Distribution::Default);
     let bias = Tensor::zeros([1, 3]);
-    let x = Tensor::random([3, 3], Distribution::Standard);
+    let x = Tensor::random([3, 3], Distribution::Default);
 
     let y = linear::<Backend>(x.clone(), weight.clone(), bias.clone());
     // y.backward() // Method backward doesn't exist