本文主要是介绍pytorch单层感知机,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
目录
- 1.单层感知机模型
- 2. 推导单层感知机梯度
- 3. 实战
1.单层感知机模型
2. 推导单层感知机梯度
公式前加了一个1/2是为了消除平方2,不加也是可以的,不会改变函数的单调性
3. 实战
- 初始化1行10列的x和w
- sigmod中x@w.t() w做了转置操作是为了将[1,10]转换成[1,1]
- torch.ones(1,1)初始化一个目标值,然后与o做MSE均方差得出loss
- 对loss做backward就计算出来w的梯度
注意:loss是一个标量
这篇关于pytorch单层感知机的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!