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作为这个样本的预测标签
桥本环奈真好看!!!!