tf.nn.softmax参数详解以及作用

tf.nn.softmax参数详解以及作用

tf.nn.softmax(logits,axis=None,name=None,dim=None)
logits:一个非空的Tensor。必须是下列类型之一:half, float32,float64
axis:将在其上执行维度softmax。默认值为-1,表示最后一个维度
name:操作的名称(可选)
dim:axis的已弃用的别名

返回:一个Tensor,与logits具有相同的类型和shape

通过Softmax回归,将logistic的预测二分类的概率的问题推广到了n分类的概率的问题。

softmax的输出向量是概率,该样本属于各个类的概率。输出的向量的每个值的大小范围为0到1。
当一个样本经过softmax层并输出一个向量,会取这个向量中值最大的那个数的index作为这个样本的预测标签

tf.nn.softmax参数详解以及作用_第1张图片桥本环奈真好看!!!!

你可能感兴趣的:(深度学习,深度学习)