pytorch_实现神经网络

  1. numpy和pytorch实现梯度下降法
    a.设定初始值
    b.求取梯度
    c.在梯度方向上进行参数的更新
  2. numpy和pytorch实现线性回归
  3. pytorch实现一个简单的神经网络

1/numpy和pytorch实现梯度下降法

在3中有体现

2/np和torch实现线性回归

在3中有体现

3/pytorch实现一个简单的神经网络

参考:https://pytorch.apachecn.org/docs/1.0/#/blitz_neural_networks_tutorial
在这里插入图片描述
使用numpy实现两层神经网络
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
从结果中可以看出,其loss在逐渐减小

使用pytorch构建两层神经网络

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

从时间上,np明显慢于torch,说明使用torch更能快速的计算出所需要的结果。

猜你喜欢

转载自blog.csdn.net/qq_42017043/article/details/89107477
今日推荐