pytorch深度学习——神经网络的基本骨架nn.Module的使用

torch.nn

查看官网，找到nn.Module

1. 在官网，Docs(上面导航栏)->pytroch->torch.nn(左边导航栏)，就可以看到这个包下面的工具
   
2. 点击Contains，可以看到Module
   
3. 点击Module，可以看到nn.Module
   
   所有的神经网络都应该继承这个类，重写forward方法，那么神经网络的input就会经过forward转换得到output
   代码示例：

import torch
from torch import nn

class MyModule(nn.Module):
    def __int__(self):
        super(MyModule, self).__int__()

    def forward(self, input):# 重写方法 将输入加一得到输出
        output = input+1
        return output

mymodule = MyModule()# 创建一个神经网络
x = torch.tensor(1)
ouput = mymodule(x)
print(ouput)

输出示例：

tensor(2)