本文介绍了softmax函数在处理神经网络输出时的作用，强调了在维度选择上，dim=0适用于每一列元素和为1，而dim=1适用于每一行元素和为1。理解这一概念对于确保模型预测概率分布的正确性至关重要。
在使用softmax之后，大家一定要把outputs输出查看是否满足每一行或者每一列值相加为1（看自己输出是一列还是一行）
（1）dim=0：对每一列的所有元素进行softmax运算，并使得每一列所有元素和为1。
（2）dim=1：对每一行的所有元素进行softmax运算，并使得每一行所有元素和为1。
softmax避坑，训练集80、90准确率，测试集一直50左右
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
