动手学深度学习8-softmax分类pytorch简洁实现
摘要:
<! TOC "定义和初始化模型" "softamx和交叉熵损失函数" "定义优化算法" "训练模型" <! /TOC 定义和初始化模型 softmax的输出层是一个全连接层,所以我们使用一个线性模块就可以,因为前面我们数据返回的每个batch的样本X的形状为(batch_size,1,28,28) 阅读全文
posted @ 2019-11-05 21:40 多一点 阅读(2632) 评论(0) 推荐(0) 编辑