Pytorch系列(二):简单求导

一、 定义一个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]])