【Pytorch】7.使用Module模块搭建简易神经网络

什么是Moudel模块

torch.nn中的module是PyTorch中用于构建神经网络模型的基本单元。它包含了各种神经网络层、激活函数、损失函数等，可以通过组合不同的module来构建复杂的神经网络模型。每个module都包含了参数和方法，可以进行前向传播和反向传播等操作。使用module可以方便地构建、训练和部署神经网络模型。

如何通过Module来搭建一个简易的神经网络

首先我们现在torch.nn的官方文档中找一下module

我们可以看到，在我们神经网络的Containers中，Module的形容文，所有神经网络模型中的基类

我们可以看到Module的使用要求为，你的模型必须是这个类的子类，并且给出了自定义模型的一个例子

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

class