Softmax 層
Softmax 類別
keras.layers.Softmax(axis=-1, **kwargs)
Softmax 激活函數層。
公式
exp_x = exp(x - max(x))
f(x) = exp_x / sum(exp_x)
範例
>>> softmax_layer = keras.layers.Softmax()
>>> input = np.array([1.0, 2.0, 1.0])
>>> result = softmax_layer(input)
>>> result
[0.21194157, 0.5761169, 0.21194157]
參數
- axis：整數或整數列表，softmax 正規化應用於其上的軸。
- **kwargs：基礎層關鍵字參數，例如
name和dtype。
呼叫參數
- inputs：softmax 層的輸入 (logits)。
- mask：與
inputs形狀相同的布林遮罩。遮罩指定 1 表示保留，0 表示遮罩。預設為None。
返回
具有與 inputs 相同形狀的 Softmax 輸出。