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

tf.nn.softmax函數(shù)

別名：

• tf.math.softmax
• tf.nn.softmax

tf.nn.softmax(
    logits,
    axis=None,
    name=None,
    dim=None
)

定義在：tensorflow/python/ops/nn_ops.py。

計算softmax激活。（棄用的參數(shù)）

有些參數(shù)已被棄用。它們將在將來的版本中刪除。更新說明：不推薦使用dim，而是使用axis。

此函數(shù)執(zhí)行相當于：

softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis)

參數(shù)：

• logits：一個非空的Tensor。必須是下列類型之一：half， float32，float64。
• axis：將在其上執(zhí)行維度softmax。默認值為-1，表示最后一個維度。
• name：操作的名稱（可選）。
• dim：axis的已棄用的別名。

返回：

一個Tensor，與logits具有相同的類型和shape。

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

• InvalidArgumentError：如果logits為空或axis超出logits的最后一個維度。