Softmax 层
Softmax 类
keras.layers.Softmax(axis=-1, **kwargs)
Softmax 激活层。
公式
exp_x = exp(x - max(x))
f(x) = exp_x / sum(exp_x)
示例
>>> softmax_layer = keras.layers.Softmax()
>>> input = np.array([1.0, 2.0, 1.0])
>>> result = softmax_layer(input)
>>> result
[0.21194157, 0.5761169, 0.21194157]
参数
- axis: 整数或整数列表，应用 Softmax 归一化的轴。
- **kwargs: 基础层关键字参数，例如
name和dtype。
调用参数
- inputs: Softmax 层的输入（logits）。
- mask: 与
inputs形状相同的布尔掩码。掩码指定 1 保留，0 掩盖。默认为None。
返回值
与 inputs 形状相同的 Softmax 输出。