mindspore.mint.cumprod

mindspore.mint.cumprod(input, dim, dtype=None)

计算输入Tensor input 沿轴 dim 的累计积，当前为实验性接口。

\[y_i = x_1 * x_2 * x_3 * ... * x_i\]

参数：

• input (Tensor) - 输入Tensor。

• dim (int) - 累积和计算的轴。

• dtype (mindspore.dtype, 可选) - 输出数据类型。如果不为None，则输入会转化为 dtype。这有利于防止数值溢出。如果为None，则输出和输入的数据类型一致。默认值： None 。

返回：

Tensor，和输入Tensor的shape相同。

异常：

• TypeError - 如果 input 不是Tensor。

• TypeError - axis 不是int。

• ValueError - 如果 dim 超出范围。

支持平台：

Ascend

样例：

>>> import numpy as np
>>> from mindspore import Tensor, mint
>>> x = Tensor(np.array([1, 2, 3], np.float32))
>>> output = mint.cumprod(x, 0)
>>> print(output)
[1. 2. 6.]