0
点赞
收藏
分享

微信扫一扫

nn.Module,利用potorch搭建一个最基本的神经网络。

对于Model这个类,

第一句:定义一个初始化函数,

第二句:从父类nn.Module中继承了父类的初始化函数,

第三和第四句都是自己写的。

class Model(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(1, 20, 5)
        self.conv2 = nn.Conv2d(20, 20, 5)
def forward(self, x):
        x = F.relu(self.conv1(x))
        return F.relu(self.conv2(x))

 对于forward的含义如下,

 

forward函数在每个子类中都需要重写, 

 

 搭建一个最基本的神经网络,有着最基本的输入和输出:

import torch
from torch import nn


class shenjingwangluo(nn.Module):
    def __init__(self):
        super().__init__()
    def forward(self,input):
        output = input + 1
        return output

shenjingwangluo_example = shenjingwangluo()
x = 1
output = shenjingwangluo_example(x)
print(output)

输出就是forward中所定义的: 

2
举报

相关推荐

0 条评论