nn.Sequential(nn.Flatten(),nn.Linear(784,10))
时间: 2023-07-22 11:05:58 浏览: 406
线性表之顺序表
这是一个简单的神经网络模型,使用了nn.Sequential()来构建模型。模型的第一层是nn.Flatten(),用于将输入数据从二维形状(28, 28)转换为一维形状(784,)。接下来是一个全连接层nn.Linear(784, 10),其中784是输入特征的维度,10是输出的类别数。这个模型可以用于手写数字识别等任务。
阅读全文