mindspore.Tensor.prod

mindspore.Tensor.prod(axis=(), keep_dims=False)[源代码]

默认情况下，使用指定维度的所有元素的乘积代替该维度的其他元素，以移除该维度。也可仅缩小该维度大小至1。 keep_dims 控制输出和输入的维度是否相同。

参数：

  • axis (Union[None, int, tuple(int), list(int)]) - 要减少的维度。默认值: ()，缩小所有维度。只允许常量值。当 axis 为int、tuple(int)或list(int)时，记Tensor的维度为dim，则其取值范围为[-dim, dim)。

  • keep_dims (bool) - 如果为True，则保留缩小的维度，大小为1。否则移除维度。默认值：False。

返回：

与输入的张量具有相同的数据类型的Tensor。

  • 如果 axis 为()，且 keep_dims 为False，则输出一个0维Tensor，表示输入Tensor中所有元素的最大值。

  • 如果 axis 为int，取值为1，并且 keep_dims 为False，则输出的shape为 \((x_0, x_2, ..., x_R)\) 。

  • 如果 axis 为tuple(int)或list(int)，取值为(1, 2)，并且 keep_dims 为False，则输出Tensor的shape为 \((x_0, x_3, ..., x_R)\) 。

异常：

  • TypeError - axis 不是以下数据类型之一：int、Tuple或List。

  • TypeError - keep_dims 不是bool类型。

  • ValueError - axis 超出范围。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3], dtype=np.float32))
>>> output = input_x.prod()
>>> print(output)
6.0