自学pytorch --nn.module

内容是搭建网络的基础框架

import torch.nn as nn
import torch.nn.function as F


class  myModel(nn.Module):
    def __init__(self):
        super(Module,self).__init__()
        self.conv1=nn.Conv2d(1,20,5)//定义一个二维卷积 参数分别表示输入的通道数，输出的通道数，核的大小
        self.conv2=nn.Conv2d(20,20,5)//输入的通道数要和前面对的上
    def forward(self,input):
        output=F.relu(self.conv1(x))//对输入的数进行什么样的处理
        output=F.relu(self.conv2(%xdel))
        return output