mindspore.ops.Cummax

class mindspore.ops.Cummax(axis)[源代码]

返回输入Tensor在指定轴上的累计最大值与其对应的索引。

更多参考详见 mindspore.ops.cummax()

参数:
  • axis (int) - 算子操作的维度,维度的大小范围是[-input.ndim, input.ndim - 1]。

输入:
  • input (Tensor) - 输入Tensor。

输出:

一个包含两个Tensor的元组(values, indices),分别表示累积最大值和对应索引。每个输出Tensor的shape和输入Tensor的shape相同。

支持平台:

GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> import mindspore.ops as ops
>>> cummax = ops.Cummax(axis=0)
>>> x = Tensor(np.array([[3, 4, 6, 10], [1, 6, 7, 9], [4, 3, 8, 7], [1, 3, 7, 9]]).astype(np.float32))
>>> output = cummax(x)
>>> print(output[0])
[[ 3.  4.  6. 10.]
 [ 3.  6.  7. 10.]
 [ 4.  6.  8. 10.]
 [ 4.  6.  8. 10.]]
>>> print(output[1])
[[0 0 0 0]
 [0 1 1 0]
 [2 1 2 0]
 [2 1 2 0]]