Learn practical skills, build real-world projects, and advance your career
Created 5 years ago
import torch
x = torch.tensor(3.)
w = torch.tensor(4., requires_grad=True)
b = torch.tensor(5., requires_grad=True)
y = w * x + b
y
tensor(17., grad_fn=<AddBackward0>)
y.backward()
print('dy/dx', x.grad)
print('dy/dw', w.grad)
print('dy/db', b.grad)
dy/dx None
dy/dw tensor(3.)
dy/db tensor(1.)