torch.logsumexp¶
- torch.logsumexp(input, dim, keepdim=False, *, out=None)¶
返回给定维度
dim
中input
张量每行的指数之和的对数。计算在数值上是稳定的。对于由 dim 给出的求和索引 和其他索引 , 结果为
如果
keepdim
为True
,则输出张量与input
大小相同,除了在维度dim
上,其大小为 1。否则,dim
将被压缩(参见torch.squeeze()
),导致输出张量少一个(或len(dim)
个)维度。- 参数
- 关键字参数
输出 (张量, 可选) – 输出张量。
示例
>>> a = torch.randn(3, 3) >>> torch.logsumexp(a, 1) tensor([1.4907, 1.0593, 1.5696]) >>> torch.dist(torch.logsumexp(a, 1), torch.log(torch.sum(torch.exp(a), 1))) tensor(1.6859e-07)