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)
还没有任何评论哟~
