一、 定义一个x输入
import torch
# 1、 requires_grad设置为True,来跟踪与它相关的计算
x = torch.ones(2, 2, requires_grad=True)
print (x)
tensor([[1., 1.],
[1., 1.]], requires_grad=True)
2、 针对张量做一个操作 y = x + 2
y = x + 2
print(y)
tensor([[3., 3.],
[3., 3.]], grad_fn=<AddBackward0>)
3、 针对y做更多的操作
z = y * y * 3
out = z.mean()
print(z, out)
tensor([[27., 27.],
[27., 27.]], grad_fn=<MulBackward0>) tensor(27., grad_fn=<MeanBackward0>)
4、 求导数
out.backward()
print(x.grad)
tensor([[4.5000, 4.5000],
[4.5000, 4.5000]])