Advertisement

log_softmax查看值与概率的关系,在数值越大时表示概率越高

阅读量:

通过分析log_softmax的源代码,可以理解其输出值与概率之间的关联。在pytorch框架中,log_softmax函数通常存在两种实现方式,分别为torch.nn.LogSoftmax()以及F.log_softmax()。

若希望快速获取结论,可直接跳转至3.小结部分。

文章目录

  • [1. F.log_softmax()源码]( "1. F.log_softmax\(\)源代码
    • [2. torch.nn.LogSoftmax()]( "2. torch.nn.LogSoftmax\(\)
    • [3. 小结]( "3. 小结")

1. F.log_softmax()源码

复制代码

python

def log_softmax(input, dim=None, _stacklevel=3, dtype=None):

复制代码
 # type: (Tensor, Optional[int], int, Optional[int]) -> Tensor

 当dim参数未被指定时,

 dim = _get_softmax_dim('log_softmax', input.dim(), _stacklevel)

 若dtype参数未被设定,

全部评论 (0)

还没有任何评论哟~