首页 > itarticle > pytorch 初探

pytorch 初探

admin 11月 13, 2020 0

初探 Pytorch

Torch -> Pytorch -> caffe2 + pytorch

静态图 VS 动态图

静态图（Tensorflow）
可变性较差，性能较好

动态图（pytorch）
可以查看中间结果

pytorch 优势

GPU 方便
自动求导方便




import torch
from torch import autograd


x = torch.tensor(1.)
a = torch.tensor(1., requires_grad=True)
b = torch.tensor(2., requires_grad=True)
c = torch.tensor(3., requires_grad=True)

y = a**2 * x + b * x + c

print('before:', a.grad, b.grad, c.grad)
grads = autograd.grad(y, [a, b, c])
print('after :', grads[0], grads[1], grads[2])