Keras 2 API 文档 / 层 API / 激活层 / Softmax 层

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

返回值

经过 Softmax 处理的输出,与 inputs 形状相同。