1.softmax
**
1.1作用
将得到的数据变为1的分布,使结果在0-1之间,方便评估和比较预测结果;
当我们通过模型
得到三个输出(O1=0.1 , O2 = 1,O3 = 10)
那么我们就选择最大为我们想要的结果。但是三个数字之间的差距太大,那么就用softmax
将三个值归一化到0-1
之间
2.softmax
公式
例如上述的三个数,y1 = exp(0.1)/( exp(0.1) + exp(1) + exp(10) )
剩下的以此类推。
在import torch.nn.functional as F
包里面softmax
的用法有dim = 0, 1
dim = 1
是,对 d
的列求和等于 1