欣喜刺猬

文章
6
资源
0
加入时间
3年0月8天

Pytorch 多层感知机

torch.nn.Parameter()将一个不可训练的tensor转换成可以训练的类型parameter,并将这个parameter绑定到这个module里面。忽略像素之间的空间结构, 我们可以将每个图像视为具有784个输入特征 和10个类的简单分类数据集。首先,我们将实现一个具有单隐藏层的多层感知机, 它包含256个隐藏单元。这里面使用Sequential搭建了网络架构,其中先将图片进行展平(nn.Flatten)然后传入线形层,在经过relu激活函数,最后使用Linear进行输出。