PyTorch

最新推荐文章于 2025-05-21 14:13:18 发布

为丽力

最新推荐文章于 2025-05-21 14:13:18 发布

阅读量107

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/qq_43106548/article/details/120442400

版权

conda install pytorch torchvision torchaudio cpuonly -c pytorch

python -m pip install --upgrade pip
python -m pip install --upgrade pip  -i https://pypi.douban.com/simple/

conda create -n ptenv python=3.6

PyTorch安装‘

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])