当前位置:   article > 正文

pytorch中logsumexp函数的用法_torch.logsumexp

torch.logsumexp

总结

  • 文章来源: LawsonAbs@CSDN
  • 详细的代码可见我的Github

这里给出一个简单的例子。
代码如下:
在这里插入图片描述
这里的 logsumexp 计算的维度是 0,那么对于上述代码的tensor(其size 是 [2,3])就会生成一个[1,3] 的结果。 比如对于上述红框中的计算结果就是
l o g ( e − 100 + e − 1 e 12 ) log(e^{-100} + e^{-1e12}) log(e100+e1e12)
其中 − 1 e 12 -1e12 1e12 就是 − 1 0 12 -10^{12} 1012。很明显上式结果值就是-100。所以这个 l o g ( ∑ i n e i ) log (\sum_{i}^{n} e^i) log(inei) 的函数结果值就是 m a x ( i ) max(i) max(i)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/451800?site
推荐阅读
相关标签
  

闽ICP备14008679号