16 lines
605 B
ReStructuredText
16 lines
605 B
ReStructuredText
mindspore.ops.prod
|
|
==================
|
|
|
|
.. py:function:: mindspore.ops.prod(input, axis=None, keep_dims=False, dtype=None)
|
|
|
|
返回tensor在指定轴上的乘积。
|
|
|
|
参数：
|
|
- **input** (Tensor[Number]) - 输入tensor。
|
|
- **axis** (Union[int, tuple(int), list(int), Tensor]) - 指定计算轴。如果为 ``None`` ，计算 `input` 中的所有元素。默认 ``None`` 。
|
|
- **keep_dims** (bool) - 输出tensor是否保留维度。默认 ``False`` 。
|
|
- **dtype** (:class:`mindspore.dtype`) - 指定数据类型。默认 ``None`` 。
|
|
|
|
返回：
|
|
Tensor
|