From 1886e0d9378e61b65ec792bbe5a19211a618667c Mon Sep 17 00:00:00 2001 From: Morten Hjorth-Jensen Date: Sun, 26 Nov 2023 15:12:46 +0100 Subject: [PATCH] Create simplegrad.py --- doc/Programs/PyTorch/simplegrad.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 doc/Programs/PyTorch/simplegrad.py diff --git a/doc/Programs/PyTorch/simplegrad.py b/doc/Programs/PyTorch/simplegrad.py new file mode 100644 index 000000000..2e8351e79 --- /dev/null +++ b/doc/Programs/PyTorch/simplegrad.py @@ -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)