pytorch神经网络搭建

在pytorch官网中找到torch.nn，torch.nn中包含神经网络一些工具。

Container神经网络容器即骨架，后面是一些要填充入Container的东西

在container的module中

import torch.nn as nn
import torch.nn.functional as F

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))

写一个类Model继承nn.Module,需重写两个方法初始化__init__():这个必须继承父类的初始化super().__init__()。forward()编写前向传播。

继承父类初始化函数便捷方式，在pycharm的Code中选Generate，然后选Override Method。