softmax python代码

以下是Python中的softmax函数实现：

import numpy as np

def softmax(x):
    """
    计算softmax函数
    :param x: 输入
    :return: softmax值
    """
    exps = np.exp(x - np.max(x))  # 防止溢出
    return exps / np.sum(exps)

在这里，我们使用NumPy库来进行数学计算。对于给定的输入x，我们首先对其进行归一化，以防止指数函数的溢出。然后，我们计算每个元素的指数，并将它们除以它们的总和，以获得softmax值。