16 lines
372 B
Python
16 lines
372 B
Python
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])
|