torch_cumprod {torch} | R Documentation |
Cumprod
Description
Cumprod
Usage
torch_cumprod(self, dim, dtype = NULL)
Arguments
self |
(Tensor) the input tensor. |
dim |
(int) the dimension to do the operation over |
dtype |
( |
cumprod(input, dim, out=NULL, dtype=NULL) -> Tensor
Returns the cumulative product 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 \times x_2\times x_3\times \dots \times x_i
Examples
if (torch_is_installed()) {
a = torch_randn(c(10))
a
torch_cumprod(a, dim=1)
}
[Package torch version 0.13.0 Index]