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

tf.nn.log_softmax函數(shù)

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

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

請參閱指南：神經(jīng)網(wǎng)絡(luò)>分類操作

計算log softmax激活.(不贊成使用)

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

對于我們擁有的每個批次 i 和類 j：

logsoftmax = logits - log(reduce_sum(exp(logits), axis))

參數(shù)：

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

返回：

一Tensor.與...有相同的類型logits.形狀相同logits.

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

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