Python tf.keras.layers.Softmax用法及代码示例

Softmax 激活函数。

继承自：Layer，Module

用法

tf.keras.layers.Softmax(
    axis=-1, **kwargs
)

不带掩码的例子：

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:softmax 层的输入或 logits。
• mask: 一个与inputs.默认为None.掩码指定 1 保留和 0 掩码。