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.]