目录
tf.nn.softmax的作用
将输入数据转换成概率值,所有概率值的和为1。多用于多分类任务(3种及以上类别数)。计算公式
tf.nn.softmax的示例
import numpy as np
import tensorflow as tf
logits = tf.convert_to_tensor([0,1,2],dtype=tf.float32,name='logits')
# <tf.Tensor: shape=(3,), dtype=float32, numpy=array([0., 1., 2.], dtype=float32)>
probs = tf.nn.softmax(logits)
# <tf.Tensor: shape=(3,), dtype=float32, numpy=array([0.09003057, 0.24472848, 0.66524094], dtype=float32)>
np.exp([0,1,2])
# a