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]]