Fix the readme.

mdribeiro · Oct 21, 2021 · 0a0e357 · 0a0e357
1 parent 99c1407
commit 0a0e357
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -111,7 +111,7 @@ fn gradient_descent() {
         // Dummy mini-batches made of zeros.
         let xs = Tensor::zeros(&[7], kind::FLOAT_CPU);
         let ys = Tensor::zeros(&[7], kind::FLOAT_CPU);
-        let loss = (my_module.forward(&xs) - ys).pow(2).sum(kind::Kind::Float);
+        let loss = (my_module.forward(&xs) - ys).pow_tensor_scalar(2).sum(kind::Kind::Float);
         opt.backward_step(&loss);
     }
 }