快捷方式

torch.logcumsumexp

torch.logcumsumexp(input, dim, *, out=None) Tensor

返回 input 元素在维度 dim 上进行指数运算后累加求和的对数。

对于由 dim 给定的求和索引 jj 和其他索引 ii,结果为

logcumsumexp(x)ij=logj=0iexp(xij)\text{logcumsumexp}(x)_{ij} = \log \sum\limits_{j=0}^{i} \exp(x_{ij})
参数
  • input (张量) – 输入张量。

  • dim (int) – 要进行运算的维度

关键字参数

out (张量, 可选) – 输出张量。

示例

>>> a = torch.randn(10)
>>> torch.logcumsumexp(a, dim=0)
tensor([-0.42296738, -0.04462666,  0.86278635,  0.94622083,  1.05277811,
         1.39202815,  1.83525007,  1.84492621,  2.06084887,  2.06844475]))

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源