这篇博客探讨了TensorFlow中的tf.nn.log_softmax API，它对softmax计算结果取log。文章通过比较log_softmax和softmax的异同，解析计算公式，并解释了其存在的意义，特别是在计算softmax交叉熵损失时的便利性。
官方链接
tf.nn.log_softmax(
logits,
axis=None,
name=None,
dim=None
)这个api实现的功能简单，就是把softmax的结果再进行log计算一遍。先来看一下它和tf.nn.softmax的异同，以及实现代码，然后再讨论一下为什么会有这个api的存在。
- 对比
两个api都调用的同一个函数来实现，不过传入的参数不同，关系如下图：
仔细对比gennnops中的log_sotfmax和softmax的实现源码，发现只有一处不同：
log_softmax:
softmax:
最低0.47元/天 解锁文章
6274
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
