mindspore.ops.cumprod

mindspore.ops.cumprod(input, dim, dtype=None)[源代码]

返回输入的元素在 dim 维度上的累积乘积。 例如，如果 input 是大小为N的向量，则结果也将是大小为N的向量（带有元素）。

\[y_i = x_1 * x_2 * x_3 * ... * x_i\]
参数：

  • input (Tensor[Number]) - 输入Tensor，必须是数字类型。\((N,*)\) ，其中 \(*\) 表示任何额外的维度。

  • dim (int) - 计算累积乘积的尺寸。只允许常量值。

  • dtype (mindspore.dtype, 可选) - 输出的数据类型。如果为None，则输出和输入的数据类型一致。默认值： None 。

返回：

Tensor，数据类型和shape与 input 相同，除非指定了 dtype。

异常：

  • TypeError - 如果 dim 不是int。

  • TypeError - 如果 dtype 无法进行转换。

  • ValueError - 如果 dim 是None。

支持平台：

Ascend GPU CPU

样例：

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