TensorFlow函數(shù)教程：tf.keras.activations.softmax

tf.keras.activations.softmax函數(shù)

tf.keras.activations.softmax(
    x,
    axis=-1
)

定義在：tensorflow/python/keras/activations.py。

Softmax激活函數(shù)。

參數(shù)：

• x：輸入張量。
• axis：整數(shù)，應用softmax標準化的軸。

返回：

張量，softmax變換的輸出。

可能引發(fā)的異常：

• ValueError：如果dim(x) == 1。