model.py 文件源码

python
阅读 23 收藏 0 点赞 0 评论 0

项目:torch_light 作者: ne7ermore 项目源码 文件源码
def log_sum_exp(input, keepdim=False):
    assert input.dim() == 2
    max_scores, _ = input.max(dim=-1, keepdim=True)
    output = input - max_scores.expand_as(input)
    return max_scores + torch.log(torch.sum(torch.exp(output), dim=-1, keepdim=keepdim))
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号