基础函数raval,Sequential,Module

raval 将一个多维数据转化为一维数据

Parameter 和Variable类似,其默认是进行求导的不需要设置参数requires_grad = True
神经网络中的参数一般不适用Variable使用nn. Parameter
使用参数torch.optim 封装了反向求导之后的梯度更新的过程。
optimizer.zero_grad() 进行梯度归零.
optimizer0step() 进行参数更新

x1 = np.linspace(1,10)
获取一个等间隔的数据,默认数量为50

x3 = np.linspace(1,10,num = 10,retstep = True)
返回结果为一个数组和间隔的大小两个值
(array([ 1., 2., 3., 4., 5., 6., 7., 8., 9., 10.]), 1.0)
nn.Sequential 函数
seq_net = nn.Sequential(
nn.Linear(2, 4),
nn.Tanh(),
nn.Linear(4, 1)
)
产生一个两层的神经网络 两个线性层和一个激活层
print(seq_net[2]) 输出每层的神经元个数
print(seq_net[0].weight) 输出每层的参数
param = seq_net.parameters() 统计神经网络中的可进行求导的参数,不能查看 仅返回一个对象,后续进行参数更新的时候可使用
print(param)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值