Softmax与分类模型 Softmax原理 softmax通过下式将输出值变换成值为正且和为1的概率分布: 因此softmax运算不改变预测类别输出。 pytorch定义softmax函数如下: def softmax(X): X_exp = X.exp() partition = X_exp.sum(dim=1