介绍

计算softmax激活函数

进行下面操作：

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

常用别名

• tf.math.softmax

语法

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

参数：

• logits：非空Tensor，必须是下面类型之一：half, float32, float64
• axis：按照哪个轴计算，默认为-1，最后一个维度

返回值：

一个Tensor，与logits具有相同的类型和shape

例子

a=tf.constant([0.1,0.02,0.3],dtype=tf.float32)
a=tf.nn.softmax(a)
print(a)

执行结果：

tf.Tensor([0.31801364 0.2935636  0.38842276], shape=(3,), dtype=float32)