Create simplegrad.py

2023-11-26 15:12:46 +01:00
parent 3cbcd82662
commit 1886e0d937
1 changed files with 10 additions and 0 deletions
@@ -0,0 +1,10 @@
+import torch
+w = torch.tensor(1.0, requires_grad = True)
+b = torch.tensor(0.5, requires_grad = True)
+x = torch.tensor([1.4])
+y = torch.tensor([2.1])
+z = torch.add(torch.mul(w,x),b)
+loss = (y-z).pow(2).sum()
+loss.backward()
+print(w.grad)
+print(b.grad)