softmax 函数在 Python 中可以用以下代码实现:
import numpy asnp
def softmax(x):
return np.exp(x) / np.sum(np.exp(x), axis=0)
上面的代码使用了 Numpy 库来实现对向量 x
进行 softmax 计算。其中,np.exp(x)
计算了 x
中每一个元素的指数,np.sum(np.exp(x), axis=0)
计算了所有指数的和,最终通过除法得到了 softmax 函数的输出。