230523

autograd.py

@@ -0,0 +1,15 @@
+import torch
+from torch import autograd
+
+device = torch.device('mps')
+
+x = torch.tensor(1.)
+a = torch.tensor(2., requires_grad=True)
+b = torch.tensor(2., requires_grad=True)
+c = torch.tensor(3., requires_grad=True)
+
+y = a ** 2 * x + b * x + c ** 3
+
+print('before:', a.grad, b.grad, c.grad)
+grads = autograd.grad(y, [a, b, c])
+print('after:', grads[0], grads[1], grads[2])