动手深度学习笔记(十七)4.3. 多层感知机的简洁实现
4. 多层感知机
4.3. 多层感知机的简洁实现
正如你所期待的,我们可以通过高级API更简洁地实现多层感知机。
import torch
from torch import nn
from d2l import torch as d2l
4.3.1. 模型
与softmax回归的简洁实现( 3.7节)相比, 唯一的区别是我们添加了2个全连接层(之前我们只添加了1个全连接层)。 第一层是隐藏层,它包含256个隐藏单元,并使用了ReLU激活函数。 第二层是输出层。