ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

torch的基础学习

2022-05-20 10:32:10  阅读:164  来源: 互联网

标签:None tensor True torch 基础 学习 grad grads


1.能带来什么

  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)       # before= None None None
grads = autograd.grad(y, [a, b, c])
print('after=', grads[0], grads[1], grads[2])  # after= tensor(2.) tensor(1.) tensor(1.)

  大量的Api,Tensor运算,神经网络

 

2.手写

    

标签:None,tensor,True,torch,基础,学习,grad,grads
来源: https://www.cnblogs.com/juncaoit/p/16291394.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有