torch.logsumexp的计算就是字面意思 但是自己实现的话发现单exp这一步输出就会出现溢出变成inf,就是无穷大 发现函数里头的小技巧是进行了平移: 参考: 机器学习 - 计算 Log-Sum-Exp 8. log_sum_exp的trick