pytorch实现逻辑回归
一.torch实现逻辑回归
二.用pytorch实现torch.nn.module
torch.nn是专门为神经网络设计的模块接口,nn构建于autograd之上,可以用来定义和运行神经网络。
定义自己的网络:
需要继承nn.Module类,并实现forward方法;
一般把网络中具有可学习参数的层放在构造函数__init__()中,
不具有可学习参数的(如Relu)可放在构造函数中,
只要在nn.Modeule的子类中定义了forward函数,backward函数就会被自动实现