Update README.md

mr-raccoon-97 · Oct 5, 2023 · c84fa19 · c84fa19
1 parent 16402f4
commit c84fa19
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -17,10 +17,10 @@ The API is currently inspired by PyTorch, with one notable difference: when you
 #include <CaberNet/CaberNet.h>
 
 int main() {
-    net::Tensor x({2,3}, false); x.fill({1,2,3,4,5,6});
-    net::Tensor w({4,3}, true); w.fill({1,2,-3,4,5,6,7,8,-9,10,11,-12});
-    net::Tensor b({1,4}, true); b.fill({1,2,3,4});
-    net::Tensor I({2,4}, false); I.fill(1);
+    net::Tensor x({2,3}, net::requires_gradient::False); x.fill({1,2,3,4,5,6});
+    net::Tensor w({4,3}, net::requires_gradient::True); w.fill({1,2,-3,4,5,6,7,8,-9,10,11,-12});
+    net::Tensor b({1,4}, net::requires_gradient::True); b.fill({1,2,3,4});
+    net::Tensor I({2,4}, net::requires_gradient::False); I.fill(1);
 
     x = net::function::linear(x,w,b);
     x = net::function::relu(x);