pytorch中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(e−100+e−1e12)
其中
−
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)
更多推荐
4
0- 0
已为社区贡献10条内容
所有评论(0)