torch.logcumsumexp

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

Возвращает логарифм кумулятивной суммы экспоненты элементов input по размерности %%%CODE_BLOCK_2%%.

Для индекса суммирования $j$, заданного dim, и других индексов $i$, результат равен:

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