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