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