神经网络-nn.Module

torcn.nn是专门为神经网络设计的模块化接口. nn构建于autograd之上，可以用来定义和运行神经网络。
nn.Module是所有网络层基类，用来管理网络属性。所有的网络模型比如说LeNet模型，包括模型的所有子模块，卷积层、池化层也是一个nn.Module类

input—>神经网络–>output
下面写个示例来复习下forward()（前向传播函数）

import torch
from torch import nn

class Lwx(nn.Module):
    def __init__(self):
        super(Lwx, self).__init__()
    def forward(self,input):
        output = input+1
        return output

lwx=Lwx()
x=torch.tensor(1.0)
y=lwx(x)
# tensor(2.)
print(y)

在上组示例中，当调用lwx(x)时，系统会自动调用forward()函数并返回
有个关于nn.module模块写的不错
Pytorch nn.Module模块详解
标记下，以后忘了自己好看