mlx.core.prod

prod(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array

在给定轴上进行乘积归约。

参数:

• a (array) – 输入数组。

• axis (int 或 list(int), 可选) – 可选的用于归约的轴。如果未指定，则默认在整个数组上进行归约。

• keepdims (bool, 可选) – 保持被归约的轴作为单维度，默认为 False。

返回值:

对应轴被归约后的输出数组。

返回类型:

array