tf.nn.softmax(logits, axis=None, name=None)
等效执行:
softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis)
Args:
logits
:一个非空的Tensor。必须是下列类型之一:Tensor half float32 float64
axis
:在该维度上执行softmax。默认值为-1,表示最后一个维度
name
:操作的名称(可选)
Returns:
与Tensor logits
具有相同的type和shape