Softmax 层
Softmax 类
tf_keras.layers.Softmax(axis=-1, **kwargs)
Softmax 激活函数。
无掩码示例
>>> inp = np.asarray([[1., 2., 1.]])
>>> layer = tf.keras.layers.Softmax()
>>> layer(inp).numpy()
array([[0.21194157, 0.5761169 , 0.21194157]], dtype=float32)
>>> mask = np.asarray([[True, False, True]], dtype=bool)
>>> layer(inp, mask).numpy()
array([[0.5, 0. , 0.5]], dtype=float32)
输入形状
任意。在模型中使用此层作为第一层时，请使用关键字参数 input_shape（整数元组，不包含样本轴）。
输出形状
与输入形状相同。
参数
- axis: 整数或整数列表，应用 softmax 归一化的轴。
调用参数
- inputs: 输入，或 softmax 层的 logits。
- mask: 与
inputs形状相同的布尔掩码。掩码指定 1 保留，0 掩盖。默认为None。
返回值
与 inputs 形状相同的 Softmax 输出。