PyTorch 神经网络基础

模型构造

在代码里 任何一个层或者神经网络都应该是Module的子类

通过继承nn.Module 可以更灵活的去自定义我们的操作

参数管理

自定义层

带参数的层需要给出输入的维度和输出的维度

读写文件