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

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

Examples

if (torch_is_installed()) {

a <- torch_randn(c(10))
torch_logcumsumexp(a, dim=1)
}
#> torch_tensor
#>  1.2812
#>  1.4475
#>  1.7365
#>  1.9937
#>  2.1047
#>  2.1927
#>  2.4261
#>  2.6353
#>  2.6576
#>  2.7507
#> [ CPUFloatType{10} ]