torch_logcumsumexp: Logcumsumexp

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Logcumsumexp

Description

Logcumsumexp

Usage

torch_logcumsumexp(self, dim)

Arguments

self	(Tensor) the input tensor.
dim	(int) the dimension to do the operation over

logcumsumexp(input, dim, *, out=None) -> Tensor

Returns the logarithm of the cumulative summation of the exponentiation of elements of input in the dimension dim.

For summation index j given by dim and other indices i, the result is

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

Examples

if (torch_is_installed()) {

a <- torch_randn(c(10))
torch_logcumsumexp(a, dim=1)
}

torch documentation built on June 7, 2023, 6:19 p.m.