PyTorch(人工神经元)

主要是熟悉一下pytorch 的nn模块。

代码如下：

1.这个unsqueeze函数主要增加了一个维度

                   

                     

2. Python中 类的定义方式：（这一块不是很理解）回头看一下python类的定义。

    继承了nn.module模块，首先定义初始化函数，进行初始化；

    nn.Linear()函数构造线性模型；

    forward函数构造神经网络前向传播的计算步骤；

                     

3. 损失函数，以及优化函数（SGD代表随机梯度下降，参数一代表神经网络的参数）

                    

4.model 就是上面的类，传入参数是inputs；

   criterion是MSE类型的对象。

  optimizer.zero_grad() 清空梯度，loss.backward()计算梯度，optimizer.step()更新梯度

 好像这里计算梯度，没有细化到计算谁的梯度。上面一篇文章是直接知道w的梯度的。

                     

 

5.结果展示