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作为这个样本的预测标签