在pytorch中使用到了python中的类,有点懵,写下来,加强记忆了!(都一样,换了个外套而已,熟悉熟悉)
这是pytorch中的一段代码。
LR继承父类nn.Module,那么父类的属性以及共有函数,LR都可以使用。
1.第一个就是LR的初始化函数,初始化的参数通常和 类的实例化是一致的。如下:
LR的初始化函数直接初始化了linear属性。
2.第二函数就是定义的前向传播函数了需要调用的
output = model(inputs) -------------model就是LR实例的一个对象
奇怪没有看到它调用forward函数,而是直接就用了。(有点奇怪)
我试了改了一下,也可以出来正确结果。
大概就是这个意思了,哈哈哈。