python定义softmax函数

以下是Python中定义softmax函数的示例代码：

import numpy as np

def softmax(z):
    exp_z = np.exp(z)
    return exp_z / np.sum(exp_z, axis=1, keepdims=True)

其中，`z`是一个numpy数组，表示softmax函数的输入。函数首先将`z`中的每个元素作为指数的底数进行指数运算，然后对所有指数的和进行归一化处理，得到每个元素的概率值。

在实现中，我们使用numpy库的`exp`函数计算指数，使用`sum`函数计算所有指数的和，`axis=1`表示对每行进行求和，`keepdims=True`表示保留维度，以便与指数数组进行归一化处理。