megengine.Tensor.prod¶
- Tensor.prod(axis=None, keepdims=False)¶
返回当前张量在给定轴
axis上所有元素求乘积后的结果。如果
axis是一个列表，该函数对这些轴上所有的元素进行规约。如果keepdims为真，输出张量和输入张量的尺寸相同（除非该维的尺度本身为 1），否则该维会被压缩（见squeeze）。- 参数
axis – 给定的轴
keepdims (
bool) – 输出张量是否需要保留原始数量的维度。
- 返回
输出张量。
实际案例
from megengine import tensor a = tensor([False, True, True, False]) b = tensor([1.0, 2.0, 3.0, 4.0]) print(a.prod().numpy()) print(b.prod().numpy())
输出：
0 24.0