神经网络的基本框架——nn.Module的使用

torch.nn 网址torch.nn — PyTorch 1.11.0 documentation

其中Containers中包含了6个模块

官方提供的模板

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

class Model(nn.Module):  #继承nn.mudule类
    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))

写一个例子看一下  将x输入神经网络中 ，输出为input+1

import torch
from torch import nn


class XiaoLei(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self,input):
        output=input+1
        return output


xiaolei = XiaoLei()
x=torch.tensor(1.0)
output = xiaolei(x)
print(output)

可以使用断点，在xiaolei = XiaoLei() 通过单步执行查看函数的执行顺序