- 博客(1)
- 收藏
- 关注
RSS订阅原创 pytorch 深度学习的一些小笔记
深度学习笔记 softmax回归适用于分类问题。它使用softmax运算输出类别的概率分布。 softmax回归是一个单层神经网络，输出个数等于分类问题中的类别个数。 交叉熵适合衡量两个概率分布的差异。 # gather在one-hot为输出的多分类问题中，可以把最大值坐标作为index传进去，然后提取到每一行的正确预测结果，这也是gather可能的一个作用。 # 传入数据 dim
2021-12-16 14:29:12
1147
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人