Python PyTorch logcumsumexp用法及代碼示例

本文簡要介紹python語言中 torch.logcumsumexp 的用法。

用法:

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

參數：

• input(Tensor) -輸入張量。

• dim(int) -進行操作的維度

關鍵字參數：

out(Tensor,可選的) -輸出張量。

返回維度 dim 中 input 的元素的求冪的累積和的對數。

對於 dim 和其他索引 i 給出的求和索引 j ，結果是

\text{logcumsumexp}(x)_{ij} = \log \sum\limits_{j=0}^{i} \exp(x_{ij})

例子：

>>> 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]))