Softmax深入理解[译] - AIUAIwww.aiuai.cn
Pytorch的交叉熵nn.CrossEntropyLoss
在训练阶段,里面是内置了softmax
操作的,因此只需要喂入原始的数据结果即可,不需要在之前再添加softmax
层。
Softmax 是非线性函数,主要用于 multi-class classification 任务中分类器的输出端.
import numpy as np
def softmax(x):
exp_x = np.exp(x) # ps 下面结果中的e+01 是科学计数法 e+01 =