Skip to content

多层感知机简单实现

1. 导入所需库

python
import torch
from torch import nn
from d2l import torch as d2l

与softmax回归的简洁实现( 09-3节)相比, 唯一的区别是我们添加了2个全连接层(之前我们只添加了1个全连接层)。 第一层是隐藏层,它包含256个隐藏单元,并使用了ReLU激活函数。 第二层是输出层。

京ICP备2024093538号-1