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
(整数元组,不包括样本轴)。
输出形状
与输入形状相同。
参数
调用参数
inputs
形状相同的布尔掩码。掩码指定 1 表示保留,0 表示屏蔽。默认为 None
。返回值
经过 Softmax 处理的输出,与 inputs
形状相同。