当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch logcumsumexp用法及代码示例


本文简要介绍python语言中 torch.logcumsumexp 的用法。

用法:

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

参数

  • input(Tensor) -输入张量。

  • dim(int) -进行操作的维度

关键字参数

out(Tensor,可选的) -输出张量。

返回维度 diminput 的元素的求幂的累积和的对数。

对于 dim 和其他索引 给出的求和索引 ,结果是

例子:

>>> 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.org大神的英文原创作品 torch.logcumsumexp。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。