mlx.core.cummax

目录

mlx.core.cummax#

cummax(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) array#

返回沿给定轴的元素的累积最大值。

参数:
  • a (array) – 输入数组

  • axis (int, optional) – 可选轴,用于计算累积最大值。如果未指定,则返回扁平化数组的累积最大值。

  • reverse (bool) – 以反向执行累积最大值。

  • inclusive (bool) – 输出的第 i 个元素是否包含输入的第 i 个元素。

返回:

输出数组。

返回类型:

array