torch.logcumsumexp

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

回傳輸入張量 input 沿著維度 dim 的元素取指數後之累加總和的對數。

對於由 dim 給定的求和索引 $j$ 和其他索引 $i$，結果為

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

參數

• input (Tensor) – 輸入張量。

• dim (int) – 要進行運算的維度

關鍵字參數

out (Tensor, optional) – 輸出張量。

範例

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