模型构造
在代码里 任何一个层或者神经网络都应该是Module的子类
通过继承nn.Module 可以更灵活的去自定义我们的操作
参数管理
自定义层
带参数的层需要给出输入的维度和输出的维度
读写文件
PyTorch 神经网络基础
发布时间 2024-01-01 14:52:21作者: 一个经常掉线的人
在代码里 任何一个层或者神经网络都应该是Module的子类
通过继承nn.Module 可以更灵活的去自定义我们的操作
带参数的层需要给出输入的维度和输出的维度