tf.nn.softmax

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

此函数执行相当于

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

参数：
● logits：非空Tensor。必须是以下类型之一：half，float32，float64。
● axis：将执行softmax维度。默认值为-1，表示最后一个维度。
● name：操作的名称（可选）。
● dim：轴的不推荐别名。

返回：张量。具有与logits相同的类型和形状。

提示：
● InvalidArgumentError：如果logits为空或轴超出logits的最后一个维度。