手动实现请参考:https://blog.csdn.net/qq_37534947/article/details/109394648
二:“利用torch.nn实现前馈神经网络解决回归、二分类、多分类任务”实验
注: 因为相比于手动实验其主要是在模型的构建中利用了torch.nn模块,其他的部分都是一样的,这里我主要介绍如何利用torch模块构建的模型。
2.1“利用torch.nn实现前馈神经网络解决回归”实验
实验过程:
2.1.1 利用torch.nn构建模型以及初始化参数
1. #实现FlattenLayer层
2. class FlattenLaye