R/gen-namespace-docs.R
, R/gen-namespace-examples.R
, R/gen-namespace.R
torch_cumsum.Rd
Cumsum
torch_cumsum(self, dim, dtype = NULL)
self | (Tensor) the input tensor. |
---|---|
dim | (int) the dimension to do the operation over |
dtype | ( |
Returns the cumulative sum of elements of input
in the dimension
dim
.
For example, if input
is a vector of size N, the result will also be
a vector of size N, with elements.
$$ y_i = x_1 + x_2 + x_3 + \dots + x_i $$
#> torch_tensor #> 1.5249 #> 3.2125 #> 3.5023 #> 4.2332 #> 3.1273 #> 2.0299 #> 2.2226 #> -0.3204 #> -1.7998 #> -2.0587 #> [ CPUFloatType{10} ]