Cumprod
Source:R/gen-namespace-docs.R
, R/gen-namespace-examples.R
, R/gen-namespace.R
torch_cumprod.Rd
Cumprod
Arguments
- self
(Tensor) the input tensor.
- dim
(int) the dimension to do the operation over
- dtype
(
torch.dtype
, optional) the desired data type of returned tensor. If specified, the input tensor is casted todtype
before the operation is performed. This is useful for preventing data type overflows. Default: NULL.
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)
}
#> torch_tensor
#> 2.0078e-01
#> -6.4478e-03
#> -5.5492e-03
#> 5.6674e-03
#> -1.5816e-03
#> -8.8559e-04
#> -8.2771e-04
#> -1.0339e-04
#> 1.9150e-04
#> 3.2690e-06
#> [ CPUFloatType{10} ]