对于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