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 掩碼。