pymc.math.prod

pymc.math.prod(input, axis=None, dtype=None, keepdims=False, acc_dtype=None, no_zeros_in_input=False)

计算张量 input 沿指定轴(轴)的乘积。

当 axis 为 None（默认值）时，乘积是在展平的张量上进行的。

完整文档请参见 tensor.elemwise.Prod。

参数:

keepdims: bool

如果设置为 True，被缩减的轴将作为尺寸为1的维度保留在结果中。使用此选项，结果将与原始张量正确广播。