pytorch 构建网络模型的方法

最新推荐文章于 2024-04-18 16:28:07 发布

qqyouhappy

最新推荐文章于 2024-04-18 16:28:07 发布

阅读量219

点赞数

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/qqyouhappy/article/details/106164274

版权

pytorch 专栏收录该内容

15 篇文章

订阅专栏

1.常规

class Logistic_regression_1(nn.Module):
def __init__(self):
super(Logistic_regression_1, self).__init__()
self.lr = nn.Linear(2, 1)
self.sm = nn.Sigmoid()

def forward(self, x):
x = self.lr(x)
x = self.sm(x)
return x

2.sequential

class Logistic_regression_2(nn.Module):
    def __init__(self):
        super(Logistic_regression_2, self).__init__()
        self.lr = torch.nn.Sequential(
            nn.Linear(2, 1)
        )
        self.sm = torch.nn.Sequential(
            nn.Sigmoid()
        )

    def forward(self, x):
        x = self.lr(x)
        x = self.sm(x)
        return x

3.add_module

class Logistic_regression_3(torch.nn.Module):
    def __init__(self):
        super(Logistic_regression_3, self).__init__()
        self.lr = torch.nn.Sequential()
        self.lr.add_module("lr",torch.nn.Linear(2, 1))
        
        self.sm = torch.nn.Sequential()
        self.sm.add_module("sm",torch.nn.Sigmoid())
       

    def forward(self, x):
        x = self.lr(x)
        x = self.sm(x)
        return x

4.OrderedDict

import torch
from collections import OrderedDict
class Logistic_regression_4(torch.nn.Module):
    def __init__(self):
        super(Logistic_regression_4, self).__init__()
        self.lr = torch.nn.Sequential(
            OrderedDict(
                [
                    ("lr", torch.nn.Linear(2, 1)),
                    
                ]
            ))

        self.sm = torch.nn.Sequential(
            OrderedDict([
                 ("sm", torch.nn.Sigmoid()),
                
            ])
        )
    
    def forward(self, x):
        x = self.lr(x)
        x = self.sm(x)
        return x

model = Net4()

print(model)