softmax 的代码示例

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

import numpy as np

def softmax(x):
    exp_x = np.exp(x)
    return exp_x / np.sum(exp_x)

x = np.array([1, 2, 3])
print(softmax(x))

这个函数接受一个一维的 Numpy 数组作为输入，然后返回一个相同大小的数组，其中每个元素都是输入数组对应元素的 softmax 值。