【Day3】3h到5h,PyTorch回顾(上)

PyTorch的安装( 学习用CPU也能接收,真正做点事情用GPU)

CPU版本安装:pip install torch1.3.0+cpu torchvision0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

GPU版本安装:pip install torch1.3.0 torchvision0.4.1 -f https://download.pytorch.org/whl/torch_stable

安装
安装成功!
安装成功,自己装的是1.6.0的

【基本操作】
tensor跟numpy有很多相似、甚至还能很快的相互转化:
empty操作随机
初始化零矩阵直接传入数据
跟numpy差不多
随机数
size
基本计算方法
view操作跟numpy的reshape差不多
view操作

索引
numpy的nparray和torch的tensor互相转换
相互转换
【自动求导机制】(反向传播全部算好了,这也是“用框架”的好处)
定义一个可求导tensor
需要求导的,进行手动定义
方法1:
方法1

方法2:(稍微麻烦一丢丢)方法2

requires_grad= True,表示需要求导
例子
b
图3
意思就是:z = wx+b
y自动需要了

进行反向传播:a.backward(),最后一层标志一下,就自动反向传播了。
反向传播
代码重复执行时,如果对梯度不清零,则会默认将梯度累加起来。因此一般要清零!

####################################
线性回归模型

构造一组输入数据X和对应的标签y
x_values=
values
x_train=
x_values
无论建立多么复杂的模型,都可以把“类”先定义出来:
class Model(nn.Module):
用def 继承过来
简单的线性回归模型的构造:
1
模型
model
发现好像不能截图粘贴了?难道这个有上限??

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值